Skip to content
Prev Previous commit
Next Next commit
fix mac error
  • Loading branch information
YqGe585 committed Dec 15, 2025
commit 992890030fa624acda2cf7346d7ff1f91210d86e
7 changes: 7 additions & 0 deletions paddle/phi/backends/all_context.h
Original file line number Diff line number Diff line change
Expand Up @@ -21,9 +21,16 @@ limitations under the License. */
// path replacement after implementing phi DeviceContext

#include "paddle/phi/backends/cpu/cpu_context.h"
#ifdef PADDLE_WITH_CUSTOM_DEVICE
#include "paddle/phi/backends/custom/custom_context.h"
#endif
#if defined(PADDLE_WITH_CUDA) || defined(PADDLE_WITH_HIP) || \
defined(PADDLE_WITH_CUSTOM_DEVICE)
#include "paddle/phi/backends/gpu/gpu_context.h"
#endif
#include "paddle/phi/backends/onednn/onednn_context.h"
#ifdef PADDLE_WITH_XPU
#include "paddle/phi/backends/xpu/xpu_context.h"
#endif

namespace phi {} // namespace phi
32 changes: 30 additions & 2 deletions python/setup.py.in
Original file line number Diff line number Diff line change
Expand Up @@ -1536,7 +1536,7 @@ if '${WITH_GPU}' == 'ON' or '${WITH_ROCM}' == 'ON':
find_files('*.h', '@PADDLE_SOURCE_DIR@/paddle/phi/backends/cpu')
)
headers += list(
find_files('*.h', '@PADDLE_SOURCE_DIR@/paddle/phi/backends/gpu')
find_files('*.h', '@PADDLE_SOURCE_DIR@/paddle/phi/backends/gpu, recursive=True')
)
headers += list(
find_files(
Expand Down Expand Up @@ -1587,7 +1587,35 @@ if '${WITH_CUSTOM_DEVICE}' == 'ON':
find_files('*.h', '@PADDLE_SOURCE_DIR@/paddle/phi/backends/custom')
)
headers += list(
find_files('*.h', '@PADDLE_SOURCE_DIR@/paddle/phi/backends/gpu')
find_files('*.h', '@PADDLE_SOURCE_DIR@/paddle/phi/backends/gpu', recursive=True)
)
headers += list(
find_files('*.h', '@PADDLE_SOURCE_DIR@/paddle/phi/backends/onednn')
)
headers += [
os.path.join(
'@PADDLE_SOURCE_DIR@/paddle/phi/backends/dynload/afs_api.h'
),
os.path.join(
'@PADDLE_SOURCE_DIR@/paddle/phi/backends/dynload/dynamic_loader.h',
),
os.path.join(
'@PADDLE_SOURCE_DIR@/paddle/phi/backends/dynload/mklml.h'
),
os.path.join(
'@PADDLE_SOURCE_DIR@/paddle/phi/backends/dynload/mklrt.h'
),
os.path.join(
'@PADDLE_SOURCE_DIR@/paddle/phi/backends/dynload/lapack.h'
),
os.path.join(
'@PADDLE_SOURCE_DIR@/paddle/phi/backends/dynload/hml.h'
),
]
else:
headers += list(
find_files('*.h', '@PADDLE_SOURCE_DIR@/paddle/phi/backends/cpu')
)
)
headers += list(
find_files('*.h', '@PADDLE_SOURCE_DIR@/paddle/phi/backends/onednn')
Expand Down
38 changes: 36 additions & 2 deletions setup.py
Original file line number Diff line number Diff line change
Expand Up @@ -2326,7 +2326,10 @@ def get_headers():
find_files('*.h', paddle_source_dir + '/paddle/phi/backends/cpu')
)
headers += list(
find_files('*.h', paddle_source_dir + '/paddle/phi/backends/gpu')
find_files(
'*.h',
paddle_source_dir + '/paddle/phi/backends/gpu, recursive=True',
)
)
headers += list(
find_files(
Expand Down Expand Up @@ -2414,7 +2417,38 @@ def get_headers():
find_files('*.h', paddle_source_dir + '/paddle/phi/backends/custom')
)
headers += list(
find_files('*.h', paddle_source_dir + '/paddle/phi/backends/gpu')
find_files(
'*.h',
paddle_source_dir + '/paddle/phi/backends/gpu, recursive=True',
)
)
headers += list(
find_files('*.h', paddle_source_dir + '/paddle/phi/backends/onednn')
)
headers += [
os.path.join(
paddle_source_dir, 'paddle/phi/backends/dynload/afs_api.h'
),
os.path.join(
paddle_source_dir,
'paddle/phi/backends/dynload/dynamic_loader.h',
),
os.path.join(
paddle_source_dir, 'paddle/phi/backends/dynload/mklml.h'
),
os.path.join(
paddle_source_dir, 'paddle/phi/backends/dynload/mklrt.h'
),
os.path.join(
paddle_source_dir, 'paddle/phi/backends/dynload/lapack.h'
),
os.path.join(
paddle_source_dir, 'paddle/phi/backends/dynload/hml.h'
),
]
else:
headers += list(
find_files('*.h', paddle_source_dir + '/paddle/phi/backends/cpu')
)
headers += list(
find_files('*.h', paddle_source_dir + '/paddle/phi/backends/onednn')
Expand Down
Loading