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

import mxnet as mx

x = mx.np.array([[0, 1], [1, 1], [2, 2]], ctx=mx.gpu())
idx = x < 2
x[idx]