[GitHub] [incubator-mxnet] leezu opened a new issue #17886: Boolean tvm operators broken on gpu

[GitBox <gi...@apache.org>]

leezu opened a new issue #17886: Boolean tvm operators broken on gpu
URL: https://github.com/apache/incubator-mxnet/issues/17886
 
 
   ## 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]
   ```

----------------------------------------------------------------
This is an automated message from the Apache Git Service.
To respond to the message, please log on to GitHub and use the
URL above to go to the specific comment.
 
For queries about this service, please contact Infrastructure at:
users@infra.apache.org


With regards,
Apache Git Services

[GitHub] [incubator-mxnet] ChaiBapchya commented on issue #17886: Boolean tvm operators broken on gpu

[GitBox <gi...@apache.org>]

ChaiBapchya commented on issue #17886:
URL: https://github.com/apache/incubator-mxnet/issues/17886#issuecomment-620670074


   Duplicate of #17840 


----------------------------------------------------------------
This is an automated message from the Apache Git Service.
To respond to the message, please log on to GitHub and use the
URL above to go to the specific comment.

For queries about this service, please contact Infrastructure at:
users@infra.apache.org