File tree Expand file tree Collapse file tree 2 files changed +4
-4
lines changed Expand file tree Collapse file tree 2 files changed +4
-4
lines changed Original file line number Diff line number Diff line change 1- # Copyright 2021-2023 , NVIDIA CORPORATION & AFFILIATES. All rights reserved.
1+ # Copyright 2021-2025 , NVIDIA CORPORATION & AFFILIATES. All rights reserved.
22#
33# Redistribution and use in source and binary forms, with or without
44# modification, are permitted provided that the following conditions
@@ -60,7 +60,7 @@ def test_bls_out_of_memory(self):
6060 input0_np , infer_response = self ._send_identity_tensor (
6161 tensor_size , self ._is_decoupled
6262 )
63- out_of_memory_message = "Failed to increase the shared memory pool size for key "
63+ out_of_memory_message = "Failed to increase the shared memory pool size"
6464
6565 if infer_response .has_error ():
6666 self .assertIn (out_of_memory_message , infer_response .error ().message ())
Original file line number Diff line number Diff line change 1- # Copyright 2021-2023 , NVIDIA CORPORATION & AFFILIATES. All rights reserved.
1+ # Copyright 2021-2025 , NVIDIA CORPORATION & AFFILIATES. All rights reserved.
22#
33# Redistribution and use in source and binary forms, with or without
44# modification, are permitted provided that the following conditions
@@ -53,7 +53,7 @@ async def test_bls_out_of_memory():
5353 tensor_size = 256 * 1024 * 1024
5454 input0_np , infer_response = await _send_identity_tensor (tensor_size , is_decoupled )
5555
56- out_of_memory_message = "Failed to increase the shared memory pool size for key "
56+ out_of_memory_message = "Failed to increase the shared memory pool size"
5757
5858 if infer_response .has_error ():
5959 if not (out_of_memory_message in infer_response .error ().message ()):
You can’t perform that action at this time.
0 commit comments