There was an error while loading. Please reload this page.
1 parent 03881b0 commit ecf4ce7Copy full SHA for ecf4ce7
torch/_inductor/cudagraph_trees.py
@@ -82,7 +82,7 @@
82
StorageDataPtr = int
83
NBytes = int
84
85
-if torch.has_cuda:
+if torch.backends.cuda.is_built():
86
from torch._C import (
87
_cuda_CUDAAllocator_AllocatorState as AllocatorState,
88
_set_cached_tensors_enabled as _set_cached_tensors_enabled,
0 commit comments