Update default CUDA version to 12.4 (#1028)

Summary: Pull Request resolved: #1028 Reviewed By: hl475 Differential Revision: D63041213
facebookincubator · Oct 30, 2024 · 0037b39 · 0037b39
1 parent c7e0101
commit 0037b39
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__(