Skip to content

Commit a0350f4

Browse files
committed
append '_cpu' tag correctly
- fix custom command to work with -B build option
1 parent 73d3e7b commit a0350f4

File tree

1 file changed

+5
-5
lines changed

1 file changed

+5
-5
lines changed

CMakeLists.txt

Lines changed: 5 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -128,15 +128,15 @@ elseif(BUILD_MPS)
128128
string(APPEND BNB_OUTPUT_NAME "_mps")
129129
add_compile_definitions(BUILD_MPS)
130130
file(MAKE_DIRECTORY "build")
131-
add_custom_command(OUTPUT "bitsandbytes/bitsandbytes.metallib"
132-
COMMAND xcrun metal -c -o "build/bitsandbytes.air" ${METAL_FILES}
133-
COMMAND xcrun metallib "build/bitsandbytes.air" -o "bitsandbytes/bitsandbytes.metallib"
131+
add_custom_command(OUTPUT "${CMAKE_SOURCE_DIR}/bitsandbytes/bitsandbytes.metallib"
132+
COMMAND xcrun metal -c -o "bitsandbytes.air" "${CMAKE_SOURCE_DIR}/${METAL_FILES}"
133+
COMMAND xcrun metallib "bitsandbytes.air" -o "${CMAKE_SOURCE_DIR}/bitsandbytes/bitsandbytes.metallib"
134134
DEPENDS "${METAL_FILES}"
135135
COMMENT "Compiling Metal kernels"
136136
VERBATIM)
137-
add_custom_target(metallib DEPENDS "bitsandbytes/bitsandbytes.metallib")
137+
add_custom_target(metallib DEPENDS "${CMAKE_SOURCE_DIR}/bitsandbytes/bitsandbytes.metallib")
138138
else()
139-
set(LIBSUFFIX "cpu")
139+
string(APPEND BNB_OUTPUT_NAME "_cpu")
140140
set(GPU_SOURCES)
141141
endif()
142142

0 commit comments

Comments
 (0)