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