We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent c6d2e02 commit 53eec81Copy full SHA for 53eec81
sgl-kernel/python/sgl_kernel/utils.py
@@ -13,6 +13,7 @@
13
# limitations under the License.
14
# ==============================================================================
15
16
+import functools
17
from typing import Dict, Tuple
18
19
import torch
@@ -41,6 +42,7 @@ def _to_tensor_scalar_tuple(x):
41
42
return (None, x)
43
44
45
+@functools.lru_cache(maxsize=1)
46
def is_hopper_arch() -> bool:
47
# Hopper arch's compute capability == 9.0
48
device = torch.cuda.current_device()
0 commit comments