We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 326d5b6 commit 154a8dcCopy full SHA for 154a8dc
onnxruntime/test/python/transformers/test_mha.py
@@ -892,7 +892,7 @@ def test_all(self):
892
# Run tests sequentially to avoid out of memory issue.
893
self.run_mha_cpu()
894
self.run_mha_cuda()
895
- self.run_lean_attention()
+ # self.run_lean_attention()
896
self.run_mha_cuda_multi_threading_default()
897
self.run_mha_cuda_multi_threading_cudnn()
898
self.run_mha_cuda_multi_threading_efficient()
0 commit comments