Skip to content
Open
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
4 changes: 3 additions & 1 deletion csrc/multidevice/symmetric_tensor.h
Original file line number Diff line number Diff line change
Expand Up @@ -71,12 +71,14 @@ class SymmetricTensor {
size_t requested_size_;
mutable bool are_remote_tensors_setup_ = false;
bool is_multicast_setup_ = false;
void* mc_ptr_{nullptr};
#if (CUDA_VERSION >= 13000)
CUmemGenericAllocationHandle mcast_handle_{};
CUdevice cu_dev_{};
void* mc_ptr_{nullptr};
CUdeviceptr mc_base_ptr_{0};
int exporter_rank_{-1};
int peer_fd_{-1};
#endif
bool is_contiguous_view_setup_ = false;
at::Tensor contiguous_view_;
};
Expand Down