File tree Expand file tree Collapse file tree 2 files changed +3
-3
lines changed
paddle/fluid/distributed/collective/deep_ep Expand file tree Collapse file tree 2 files changed +3
-3
lines changed Original file line number Diff line number Diff line change @@ -208,7 +208,7 @@ int Buffer::get_root_rdma_rank(bool global) const {
208
208
209
209
int Buffer::get_local_device_id () const { return device_id; }
210
210
211
- #if !defined(PADDLE_ON_INFERENCE) && !defined( PADDLE_NO_PYTHON)
211
+ #ifndef PADDLE_NO_PYTHON
212
212
pybind11::bytearray Buffer::get_local_ipc_handle () const {
213
213
return {ipc_handles[nvl_rank].reserved , CUDA_IPC_HANDLE_SIZE};
214
214
}
Original file line number Diff line number Diff line change 24
24
#undef NDEBUG
25
25
#endif
26
26
27
- #if !defined(PADDLE_ON_INFERENCE) && !defined( PADDLE_NO_PYTHON)
27
+ #ifndef PADDLE_NO_PYTHON
28
28
#include < pybind11/pybind11.h>
29
29
#include < pybind11/pytypes.h>
30
30
#endif
@@ -122,7 +122,7 @@ struct Buffer {
122
122
123
123
int get_local_device_id () const ;
124
124
125
- #if !defined(PADDLE_ON_INFERENCE) && !defined( PADDLE_NO_PYTHON)
125
+ #ifndef PADDLE_NO_PYTHON
126
126
pybind11::bytearray get_local_ipc_handle () const ;
127
127
128
128
pybind11::bytearray get_local_nvshmem_unique_id () const ;
You can’t perform that action at this time.
0 commit comments