You signed in with another tab or window. Reload to refresh your session.You signed out in another tab or window. Reload to refresh your session.You switched accounts on another tab or window. Reload to refresh your session.Dismiss alert
{{ message }}
This repository has been archived by the owner on Nov 17, 2023. It is now read-only.
For example, less_scalar_gpufloat32_2bool_2 is missing causing mxnet to crash.
Works fine for cpu.
Error Message
Traceback (most recent call last):
File "/home/ubuntu/test.py", line 4, in <module>
x[x < 2]
File "/home/ubuntu/src/mxnet-master/python/mxnet/numpy/multiarray.py", line 729, in __getitem__
return self._get_np_boolean_indexing(key_before_expaned)
File "/home/ubuntu/src/mxnet-master/python/mxnet/numpy/multiarray.py", line 490, in _get_np_boolean_indexing
after_mask = _reshape_view(_npi.boolean_mask(data, mask), -1, *remaining_shapes)
File "<string>", line 53, in boolean_mask
File "/home/ubuntu/src/mxnet-master/python/mxnet/_ctypes/ndarray.py", line 82, in _imperative_invoke
check_call(_LIB.MXImperativeInvokeEx(
File "/home/ubuntu/src/mxnet-master/python/mxnet/base.py", line 246, in check_call
raise get_last_ffi_error()
mxnet.base.MXNetError: Traceback (most recent call last):
File "../3rdparty/tvm/src/runtime/module.cc", line 123
File "../3rdparty/tvm/src/runtime/library_module.cc", line 91
TVMError: Check failed: ret == 0 (-1 vs. 0) : Check failed: f != nullptr: Cannot find function less_scalar_gpufloat32_2bool_2_kernel0 in the imported modules or global registry
To Reproduce
Steps to reproduce
Compile MXNet with USE_TVMOP=1.
Run
import mxnet as mx
x = mx.np.array([[0, 1], [1, 1], [2, 2]], ctx=mx.gpu())
idx = x < 2
x[idx]
The text was updated successfully, but these errors were encountered:
Description
For example,
less_scalar_gpufloat32_2bool_2
is missing causing mxnet to crash.Works fine for cpu.
Error Message
To Reproduce
Steps to reproduce
Compile MXNet with
USE_TVMOP=1
.Run
The text was updated successfully, but these errors were encountered: