diff --git a/torch/testing/_internal/distributed/nn/api/remote_module_test.py b/torch/testing/_internal/distributed/nn/api/remote_module_test.py index 93b81df60ec099..5f68547980abcb 100644 --- a/torch/testing/_internal/distributed/nn/api/remote_module_test.py +++ b/torch/testing/_internal/distributed/nn/api/remote_module_test.py @@ -616,7 +616,7 @@ def test_invalid_devices(self): if TEST_WITH_ROCM: errorString = (r"HIP error: invalid device ordinal\n" r"HIP kernel errors might be asynchronously reported at some other API call, so the stacktrace below might be incorrect.\n" - r"For debugging consider passing AMD_SERIALIZE_KERNEL=3.") + r"For debugging consider passing AMD_SERIALIZE_KERNEL=3") else: errorString = r"CUDA error: invalid device ordinal" with self.assertRaisesRegex(