python - CuDF apply_chunks を使用した例外 - サポートされていない NumPy 関数 'numpy.ones_like' の使用またはサポートされていない関数の使用

Question

Numba の jit 最適化コード内から numpy を使用しようとしていますが、numpy.ones_like のような標準的な numpy 操作を実行しようとするとエラーが発生します。

ドキュメント リンク: Numba 0.46。

編集:メソッド「calc_method」は、直接呼び出すと正常に動作し、apply_chunks 内から使用すると失敗します。したがって、おそらく Numba 自体の問題ではなく、cudf.apply_chunks の使用方法です。

コード：

import numba
from numba import jit
import pandas as pd
import numpy as np

print(numba.__version__)

@jit(nopython=True)
def calc_method(a,b):
    a1 = np.float64(a)
    b1 = np.float64(b)
    abc = (a1, np.ones_like(b1))
    abc_ht = np.hstack(abc)
    return abc_ht

def calculate(cudf_df: cudf, size_of_row: int):       
    return cudf_df.apply_chunks(calc_method, incols=['a', 'b'], outcols=dict(), chunks=size_of_row)

df = pd.DataFrame({'a': [1, 2, 3, 4, 5, 6, 7, 8], 'b': [11, 12, 13, 14, 15, 16, 17, 18]})
cudf_df = cudf.DataFrame.from_pandas(df)
a, b = calculate(cudf_df, 4)

エラー：

TypingError                               Traceback (most recent call last)
<ipython-input-38-ad56fb75bc4a> in <module>
----> 1 a, b = calculate(cudf_df, 4)

TypingError: Failed in nopython mode pipeline (step: nopython frontend)
Invalid use of Function(<numba.cuda.compiler.DeviceFunctionTemplate object at 0x7fa78521b550>) with argument(s) of type(s): (array(int64, 1d, A), array(int64, 1d, A))
 * parameterized
In definition 0:
    TypingError: Failed in nopython mode pipeline (step: nopython frontend)
Use of unsupported NumPy function 'numpy.ones_like' or unsupported use of the function.

File "<ipython-input-37-97f7d707ba81>", line 9:
def calc_method(a,b):
    <source elided>
    b1 = np.float64(b)
    abc = (a1, np.ones_like(b1))
    ^

上記の例で何が間違っているのか誰か教えてもらえますか? 前もって感謝します。

np.hstack でも同様のエラーが発生します

注: これは、問題を再現するための単純化された例です。