From becbef4fac4da4c314333ac1aecb8f5482b125c9 Mon Sep 17 00:00:00 2001 From: cmdr2 Date: Tue, 18 Apr 2023 17:36:52 +0530 Subject: [PATCH] Include ROCm in the list of allowed versions --- scripts/check_modules.py | 1 + 1 file changed, 1 insertion(+) diff --git a/scripts/check_modules.py b/scripts/check_modules.py index 0abf754e..fff51aa8 100644 --- a/scripts/check_modules.py +++ b/scripts/check_modules.py @@ -94,6 +94,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.4.2" for v in module_versions) return allowed_versions