Skip to content

Commit d8ca182

Browse files
committed
Silence profiler warnings in FAv4 benchmark
1 parent c725751 commit d8ca182

1 file changed

Lines changed: 5 additions & 0 deletions

File tree

benchmarks/fav4.py

Lines changed: 5 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -6,6 +6,7 @@
66

77
import csv
88
import itertools
9+
import warnings
910
from dataclasses import dataclass, asdict
1011
from typing import Literal
1112

@@ -20,6 +21,10 @@
2021

2122
from transformer_nuggets.utils.benchmark import benchmark_cuda_function_in_microseconds
2223

24+
warnings.filterwarnings("ignore", message=".*SyncActivityProfilerHandler.*")
25+
warnings.filterwarnings("ignore", message=".*profiler_start.*")
26+
warnings.filterwarnings("ignore", message=".*profiler_stop.*")
27+
2328
try:
2429
from flash_attn.cute import flash_attn_func
2530

0 commit comments

Comments
 (0)