From 023b78d1c9411ac93863aeaee0f94c47a28d18e1 Mon Sep 17 00:00:00 2001 From: cmdr2 Date: Thu, 20 Apr 2023 17:46:34 +0530 Subject: [PATCH] Allow rocm5.2 --- scripts/check_modules.py | 1 + 1 file changed, 1 insertion(+) diff --git a/scripts/check_modules.py b/scripts/check_modules.py index f1f22d32..89297d69 100644 --- a/scripts/check_modules.py +++ b/scripts/check_modules.py @@ -99,6 +99,7 @@ def include_cuda_versions(module_versions: tuple) -> tuple: allowed_versions = tuple(module_versions) allowed_versions += tuple(f"{v}+cu116" for v in module_versions) allowed_versions += tuple(f"{v}+cu117" for v in module_versions) + allowed_versions += tuple(f"{v}+rocm5.2" for v in module_versions) allowed_versions += tuple(f"{v}+rocm5.4.2" for v in module_versions) return allowed_versions