Skip to content

Commit b76ce7a

Browse files
ngimelpruthvistony
authored andcommitted
[Symmetric memory] set handle type for ROCm (pytorch#161741)
Fixes pytorch#161722 Pull Request resolved: pytorch#161741 Approved by: https://github.com/kwen2501
1 parent db49466 commit b76ce7a

File tree

1 file changed

+1
-0
lines changed

1 file changed

+1
-0
lines changed

torch/csrc/distributed/c10d/symm_mem/CUDASymmetricMemory.cu

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -454,6 +454,7 @@ void* CUDASymmetricMemoryAllocator::alloc(
454454
C10_CUDA_DRIVER_CHECK(status);
455455
456456
#elif defined(USE_ROCM)
457+
handle_type_ = Expandable_Segments_Handle_Type::POSIX_FD;
457458
hipMemAllocationProp prop = {};
458459
prop.type = hipMemAllocationTypePinned;
459460
prop.location.type = hipMemLocationTypeDevice;

0 commit comments

Comments
 (0)