Update default CUDA version to 12.4

Differential Revision: D63041213
facebookincubator · Sep 23, 2024 · 61aa26b · 61aa26b
1 parent 437b48a
commit 61aa26b
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/python/aitemplate/backend/cuda/target_def.py b/python/aitemplate/backend/cuda/target_def.py
@@ -338,6 +338,8 @@ def __init__(self, arch="80", remote_cache_bytes=None, **kwargs):
                 FBCUDA.nvcc_option_json = json.load(nvcc_option_json)
         self.nvcc_options_json = FBCUDA.nvcc_option_json
         cuda_version = self.nvcc_option_json.get("cuda_version", None)
+        if cuda_version is None:
+            _LOGGER.warning("CUDA version is not specified in nvcc_option_json")
 
         self.remote_cache_bytes = remote_cache_bytes
         super().__init__(