File tree Expand file tree Collapse file tree 2 files changed +8
-1
lines changed Expand file tree Collapse file tree 2 files changed +8
-1
lines changed Original file line number Diff line number Diff line change @@ -158,6 +158,12 @@ add_custom_command(
158
158
${CMAKE_CURRENT_BINARY_DIR} /python/paddle_custom_device/npu/passes
159
159
COMMAND ${CMAKE_COMMAND} -E copy_if_different ${CMAKE_SOURCE_DIR} /passes/*
160
160
${CMAKE_CURRENT_BINARY_DIR} /python/paddle_custom_device/npu/passes
161
+ COMMAND ${CMAKE_COMMAND} -E make_directory
162
+ ${CMAKE_CURRENT_BINARY_DIR} /python/paddle_custom_device/npu/profile
163
+ COMMAND ${CMAKE_COMMAND} -E remove -f
164
+ ${CMAKE_CURRENT_BINARY_DIR} /python/paddle_custom_device/npu/profile
165
+ COMMAND ${CMAKE_COMMAND} -E copy_if_different ${CMAKE_SOURCE_DIR} /profile/*
166
+ ${CMAKE_CURRENT_BINARY_DIR} /python/paddle_custom_device/npu/profile
161
167
COMMAND ${Python_EXECUTABLE} ${CMAKE_CURRENT_BINARY_DIR} /setup.py bdist_wheel
162
168
DEPENDS ${CUSTOM_NPU_NAME}
163
169
COMMENT "Packing whl packages------>>>" )
Original file line number Diff line number Diff line change @@ -114,7 +114,8 @@ def main():
114
114
packages= [
115
115
'paddle_custom_device',
116
116
'paddle_custom_device.npu',
117
- 'paddle_custom_device.npu.passes'
117
+ 'paddle_custom_device.npu.passes',
118
+ 'paddle_custom_device.npu.profile'
118
119
],
119
120
include_package_data=True,
120
121
package_data = {
You can’t perform that action at this time.
0 commit comments