Skip to content

Commit

Permalink
Fix lint
Browse files Browse the repository at this point in the history
  • Loading branch information
xuzhao9 committed Nov 19, 2024
1 parent f2963f1 commit da326da
Showing 1 changed file with 2 additions and 0 deletions.
2 changes: 2 additions & 0 deletions tritonbench/operators/gemm/operator.py
Original file line number Diff line number Diff line change
Expand Up @@ -24,12 +24,14 @@

from .kernels import matmul as kernels
from .partition_k import matmul_partition_k

try:
from .persistent_matmul import (
matmul_persistent,
matmul_tma_persistent,
matmul_tma_persistent_cached,
)

HAS_PRESISTENT = True
except ModuleNotFoundError:
HAS_PRESISTENT = False
Expand Down

0 comments on commit da326da

Please sign in to comment.