From 0037b399d480b05a5a3e8cc68d88b7c614dc77cd Mon Sep 17 00:00:00 2001 From: Han Zhu Date: Wed, 30 Oct 2024 16:11:58 -0700 Subject: [PATCH] Update default CUDA version to 12.4 (#1028) Summary: Pull Request resolved: https://github.com/facebookincubator/AITemplate/pull/1028 Reviewed By: hl475 Differential Revision: D63041213 --- python/aitemplate/backend/cuda/target_def.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/python/aitemplate/backend/cuda/target_def.py b/python/aitemplate/backend/cuda/target_def.py index 1523c748f..04d51af9a 100644 --- 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__(