Closed hubertlu-tw closed 1 year ago
With Kk's help, we found a bug in fused_dense_cuda on ROCm. It failed the unit test of fused_dense on ROCm before this fix.
Steps to reproduce: $ cd apex/contrib/test/fused_dense $ pytest test_fused_dense.py
With Kk's help, we found a bug in fused_dense_cuda on ROCm.
It failed the unit test of fused_dense on ROCm before this fix.
Steps to reproduce: $ cd apex/contrib/test/fused_dense $ pytest test_fused_dense.py