diff --git a/test/pytorch_test_base.py b/test/pytorch_test_base.py index 58d75e7c92e5..49ee9817161a 100644 --- a/test/pytorch_test_base.py +++ b/test/pytorch_test_base.py @@ -285,6 +285,7 @@ 'test_block_diag_scipy', # failed to handle np.complex128 as input to tensor. 'test_remainder_fmod_large_dividend_xla', # precision, remainder with 1e9 gives incorrect answer 'test_logical_not_out_xla', # constant with type f16 and f64 is not supported + 'test_scatter_add_non_unique_index_xla', # Ran out of memory in memory space vmem }, # test_indexing.py