@TccccD it should be sufficient to run the test on any GPU environment to
reproduce the failure. The CI has some special setup to decide if to run on CPU
or GPU. For your use-case, you may just want to change
``` python
def check_l2_normalization(in_shape, mode, dtype, norm_eps=1e-10):
ctx = default_context()
...
```
to
``` python
def check_l2_normalization(in_shape, mode, dtype, norm_eps=1e-10):
ctx = mx.gpu(0)
...
```
Please let me know if that doesn't help or if you have any other questions.
[ Full content available at:
https://github.com/apache/incubator-mxnet/pull/12287 ]
This message was relayed via gitbox.apache.org for [email protected]
