skip randomly

ROCm · Sep 3, 2024 · 5cc4d06 · 5cc4d06
1 parent 8110e1d
commit 5cc4d06
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/tests/test_flash_attn.py b/tests/test_flash_attn.py
@@ -1,4 +1,5 @@
 import math
+import random
 
 import pytest
 import torch
@@ -20,9 +21,12 @@
 def is_power_of_2(n):
     return n > 0 and (n & (n - 1)) == 0
 
+random.seed(42)
+
 def skip_config(**kwargs):
     if 'd' in kwargs:
-        return not is_power_of_2(kwargs['d'])
+        # return not is_power_of_2(kwargs['d'])
+        return random.random() < 0.20
     return False
 
 def is_hip():