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 1f7a57f commit bbb1d1aCopy full SHA for bbb1d1a
test/regressions/test_tril.py
@@ -7,6 +7,7 @@ class TestSimpleBinary(TestCase):
7
def test_tril(self, dtype=torch.bool):
8
max_seq_length = 131072
9
with torch.device("xpu"):
10
+ torch.xpu.empty_cache()
11
a = torch.ones(max_seq_length, max_seq_length, dtype=torch.bool)
12
causal_mask = torch.tril(a)
13
torch.xpu.synchronize()
0 commit comments