Skip to content

Commit bfc47bf

Browse files
szyszyzysfacebook-github-bot
authored andcommitted
Add CPP version of bitpacking. (#2725)
Summary: Add cpp implementation of bitpacking functions (not rely on specific hardwares). Differential Revision: D79456037
1 parent 6cfa477 commit bfc47bf

File tree

15 files changed

+2155
-0
lines changed

15 files changed

+2155
-0
lines changed

torchao/experimental/kernels/cpu/aarch64/tests/CMakeLists.txt

Lines changed: 9 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -128,6 +128,14 @@ target_link_libraries(
128128
dep
129129
)
130130

131+
add_executable(test_bitpack_fallback_compatibility test_bitpack_fallback_compatibility.cpp)
132+
target_link_libraries(
133+
test_bitpack_fallback_compatibility
134+
PRIVATE
135+
GTest::gtest_main
136+
dep
137+
)
138+
131139
include(GoogleTest)
132140
gtest_discover_tests(test_quantization)
133141
gtest_discover_tests(test_reduction)
@@ -137,3 +145,4 @@ gtest_discover_tests(test_embedding)
137145
gtest_discover_tests(test_weight_packing)
138146
gtest_discover_tests(test_qmatmul)
139147
gtest_discover_tests(test_lut)
148+
gtest_discover_tests(test_bitpack_fallback_compatibility)

torchao/experimental/kernels/cpu/aarch64/tests/build_and_run_tests.sh

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -62,3 +62,4 @@ ${CMAKE_OUT}/test_embedding
6262
${CMAKE_OUT}/test_weight_packing
6363
${CMAKE_OUT}/test_qmatmul
6464
${CMAKE_OUT}/test_lut
65+
${CMAKE_OUT}/test_bitpack_fallback_compatibility

0 commit comments

Comments
 (0)