Skip to content

Commit d8a8250

Browse files
authored
solve missing cuda quantize lib target
Differential Revision: D110990636 Pull Request resolved: #20781
1 parent a9da19c commit d8a8250

1 file changed

Lines changed: 8 additions & 2 deletions

File tree

backends/cuda/TARGETS

Lines changed: 8 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -3,9 +3,15 @@ load("@fbsource//xplat/executorch/build:runtime_wrapper.bzl", "runtime")
33
oncall("executorch")
44

55
runtime.python_library(
6-
name = "int4_dispatch",
6+
name = "quantize_op_dispatch",
77
srcs = [
8-
"int4_dispatch.py",
8+
"coalesced_int4_tensor.py",
9+
"dp4a_planar_int6_tensor.py",
10+
"quantize_op_dispatch/__init__.py",
11+
"quantize_op_dispatch/_library.py",
12+
"quantize_op_dispatch/int4_dispatch.py",
13+
"quantize_op_dispatch/int6_dispatch.py",
14+
"quantize_op_dispatch/int8_dispatch.py",
915
],
1016
visibility = [
1117
"//executorch/...",

0 commit comments

Comments
 (0)