[GitHub] leezu commented on issue #12287: Add stable nrm2 for L2_normalization

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

leezu commented on issue #12287: Add stable nrm2 for L2_normalization
URL: https://github.com/apache/incubator-mxnet/pull/12287#issuecomment-417840927
 
 
   @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.

----------------------------------------------------------------
This is an automated message from the Apache Git Service.
To respond to the message, please log on 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