diff --git a/fbgemm_gpu/test/test_utils.py b/fbgemm_gpu/test/test_utils.py index 5d56bad9a7..a845195c56 100644 --- a/fbgemm_gpu/test/test_utils.py +++ b/fbgemm_gpu/test/test_utils.py @@ -294,7 +294,7 @@ def decorator(testfn: Callable) -> Any: @wraps(testfn) # pyre-fixme[3]: Return annotation cannot be `Any`. def wrapper(*args: Any, **kwargs: Any) -> Any: - ROCM_VERSION_FILEPATH = "/opt/rocm/.info/version-dev" + ROCM_VERSION_FILEPATH = "/opt/rocm/.info/version" if TEST_WITH_ROCM: # Fail if ROCm version file is missing. if not os.path.isfile(ROCM_VERSION_FILEPATH):