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.
2
2
#
3
3
# Redistribution and use in source and binary forms, with or without
4
4
# modification, are permitted provided that the following conditions
@@ -60,7 +60,7 @@ def test_bls_out_of_memory(self):
60
60
input0_np , infer_response = self ._send_identity_tensor (
61
61
tensor_size , self ._is_decoupled
62
62
)
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"
64
64
65
65
if infer_response .has_error ():
66
66
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.
2
2
#
3
3
# Redistribution and use in source and binary forms, with or without
4
4
# modification, are permitted provided that the following conditions
@@ -53,7 +53,7 @@ async def test_bls_out_of_memory():
53
53
tensor_size = 256 * 1024 * 1024
54
54
input0_np , infer_response = await _send_identity_tensor (tensor_size , is_decoupled )
55
55
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"
57
57
58
58
if infer_response .has_error ():
59
59
if not (out_of_memory_message in infer_response .error ().message ()):
You can’t perform that action at this time.
0 commit comments