From ff7e939a3f29ecca42e1353b977e9d46b34537a2 Mon Sep 17 00:00:00 2001 From: Paikov Iurii Date: Wed, 2 Oct 2024 14:41:32 +0000 Subject: [PATCH] Fixed error string assertion in test_invalid_devices --- .../testing/_internal/distributed/nn/api/remote_module_test.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) 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(