Skip to content

Commit d551103

Browse files
committed
fix(cutile): validate the imported tile module
1 parent 5444c47 commit d551103

1 file changed

Lines changed: 1 addition & 1 deletion

File tree

deepmd/pt_expt/kernels/cutile/common.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -76,7 +76,7 @@
7676
try:
7777
from cuda import tile as _cuda_tile
7878

79-
CUTILE_AVAILABLE = True
79+
CUTILE_AVAILABLE = hasattr(_cuda_tile, "kernel")
8080
del _cuda_tile
8181
except ImportError: # pragma: no cover - exercised only without cuda.tile
8282
CUTILE_AVAILABLE = False

0 commit comments

Comments
 (0)