1 parent c725751 commit d8ca182Copy full SHA for d8ca182
1 file changed
benchmarks/fav4.py
@@ -6,6 +6,7 @@
6
7
import csv
8
import itertools
9
+import warnings
10
from dataclasses import dataclass, asdict
11
from typing import Literal
12
@@ -20,6 +21,10 @@
20
21
22
from transformer_nuggets.utils.benchmark import benchmark_cuda_function_in_microseconds
23
24
+warnings.filterwarnings("ignore", message=".*SyncActivityProfilerHandler.*")
25
+warnings.filterwarnings("ignore", message=".*profiler_start.*")
26
+warnings.filterwarnings("ignore", message=".*profiler_stop.*")
27
+
28
try:
29
from flash_attn.cute import flash_attn_func
30
0 commit comments