Skip to content

Commit a7c2c8c

Browse files
authored
[None][fix] Allow multi-threaded copy for GDRCopy wrapper (#8535)
Signed-off-by: Dongxu Yang <[email protected]>
1 parent 77fa5df commit a7c2c8c

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

cpp/tensorrt_llm/runtime/moeLoadBalancer/gdrwrap.cpp

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -201,13 +201,13 @@ void driver_get_version(gdr_t g, int* major, int* minor)
201201
int copy_to_mapping(gdr_mh_t handle, void* map_d_ptr, void const* h_ptr, size_t size)
202202
{
203203
CHECK_INITIALIZED();
204-
return GDRLOCKCALL(gdr_internal_copy_to_mapping(handle, map_d_ptr, h_ptr, size));
204+
return gdr_internal_copy_to_mapping(handle, map_d_ptr, h_ptr, size);
205205
}
206206

207207
int copy_from_mapping(gdr_mh_t handle, void* h_ptr, void const* map_d_ptr, size_t size)
208208
{
209209
CHECK_INITIALIZED();
210-
return GDRLOCKCALL(gdr_internal_copy_from_mapping(handle, h_ptr, map_d_ptr, size));
210+
return gdr_internal_copy_from_mapping(handle, h_ptr, map_d_ptr, size);
211211
}
212212

213213
void gdrCudaMalloc(void** ptr, void** devPtr, size_t mapSize, GdrMemDesc** memDesc, gdr_t handle)

0 commit comments

Comments
 (0)