Skip to content
This repository has been archived by the owner on Nov 17, 2023. It is now read-only.

Boolean tvm operators broken on gpu #17886

Closed
leezu opened this issue Mar 23, 2020 · 1 comment
Closed

Boolean tvm operators broken on gpu #17886

leezu opened this issue Mar 23, 2020 · 1 comment
Assignees
Labels

Comments

@leezu
Copy link
Contributor

leezu commented Mar 23, 2020

Description

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]
@leezu leezu added the Bug label Mar 23, 2020
@ChaiBapchya
Copy link
Contributor

Duplicate of #17840

@leezu leezu closed this as completed Apr 29, 2020
leezu added a commit that referenced this issue May 1, 2020
AntiZpvoh pushed a commit to AntiZpvoh/incubator-mxnet that referenced this issue Jul 6, 2020
Sign up for free to subscribe to this conversation on GitHub. Already have an account? Sign in.
Labels
Projects
None yet
Development

No branches or pull requests

3 participants