Skip to content

Commit 5d909ca

Browse files
zhuhan0facebook-github-bot
authored andcommitted
Update default CUDA version to 12.4 (#1028)
Summary: Pull Request resolved: #1028 Reviewed By: wenxin0319 Differential Revision: D63041213 fbshipit-source-id: eaaa3db1655e880c92cb8c7ebcf1e3eb8b227ff0
1 parent f711356 commit 5d909ca

1 file changed

Lines changed: 2 additions & 0 deletions

File tree

python/aitemplate/backend/cuda/target_def.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -341,6 +341,8 @@ def __init__(self, arch="80", remote_cache_bytes=None, **kwargs):
341341
FBCUDA.nvcc_option_json = json.load(nvcc_option_json)
342342
self.nvcc_options_json = FBCUDA.nvcc_option_json
343343
cuda_version = self.nvcc_option_json.get("cuda_version", None)
344+
if cuda_version is None:
345+
_LOGGER.warning("CUDA version is not specified in nvcc_option_json")
344346

345347
self.remote_cache_bytes = remote_cache_bytes
346348
super().__init__(

0 commit comments

Comments
 (0)