1 parent 5444c47 commit d551103Copy full SHA for d551103
1 file changed
deepmd/pt_expt/kernels/cutile/common.py
@@ -76,7 +76,7 @@
76
try:
77
from cuda import tile as _cuda_tile
78
79
- CUTILE_AVAILABLE = True
+ CUTILE_AVAILABLE = hasattr(_cuda_tile, "kernel")
80
del _cuda_tile
81
except ImportError: # pragma: no cover - exercised only without cuda.tile
82
CUTILE_AVAILABLE = False
0 commit comments