Skip to content

Commit 9ab291f

Browse files
authored
Add new ops to CMakeLists.txt
Differential Revision: D77631378 Pull Request resolved: #2647
1 parent a5f6aff commit 9ab291f

File tree

1 file changed

+5
-1
lines changed

1 file changed

+5
-1
lines changed

torchao/experimental/CMakeLists.txt

Lines changed: 5 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -116,6 +116,8 @@ if(TORCHAO_BUILD_ATEN_OPS)
116116
ops/embedding_xbit/op_embedding_xbit_aten.cpp
117117
ops/linear_8bit_act_xbit_weight/linear_8bit_act_xbit_weight.cpp
118118
ops/linear_8bit_act_xbit_weight/op_linear_8bit_act_xbit_weight_aten.cpp
119+
ops/groupwise_lowbit_weight_lut/groupwise_lowbit_weight_lut.cpp
120+
ops/groupwise_lowbit_weight_lut/op_groupwise_lowbit_weight_lut_aten.cpp
119121
)
120122
list(TRANSFORM _torchao_op_srcs_aten PREPEND "${CMAKE_CURRENT_SOURCE_DIR}/")
121123
add_library(torchao_ops_aten SHARED ${_torchao_op_srcs_aten})
@@ -161,7 +163,9 @@ if(TORCHAO_BUILD_EXECUTORCH_OPS)
161163
ops/embedding_xbit/op_embedding_xbit_executorch.cpp
162164
ops/linear_8bit_act_xbit_weight/linear_8bit_act_xbit_weight.cpp
163165
ops/linear_8bit_act_xbit_weight/op_linear_8bit_act_xbit_weight_executorch.cpp
164-
)
166+
ops/groupwise_lowbit_weight_lut/groupwise_lowbit_weight_lut.cpp
167+
ops/groupwise_lowbit_weight_lut/op_groupwise_lowbit_weight_lut_executorch.cpp)
168+
165169
list(TRANSFORM _torchao_op_srcs_executorch PREPEND "${CMAKE_CURRENT_SOURCE_DIR}/")
166170
add_library(torchao_ops_executorch STATIC ${_torchao_op_srcs_executorch})
167171
target_link_torchao_parallel_backend(torchao_ops_executorch executorch)

0 commit comments

Comments
 (0)