Skip to content

Commit 58f5e85

Browse files
committed
Update
[ghstack-poisoned]
2 parents 41b1113 + 7d3b20f commit 58f5e85

3 files changed

Lines changed: 42 additions & 3 deletions

File tree

backends/vulkan/test/op_tests/conv2d_gemm_dynamic_test.cpp

Lines changed: 0 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -247,7 +247,6 @@ std::vector<float> conv2d_ref_xnnpack(
247247
out_min,
248248
out_max,
249249
/*flags=*/0,
250-
/*code_cache=*/nullptr,
251250
/*weights_cache=*/nullptr,
252251
&op);
253252
if (create_status != xnn_status_success || op == nullptr) {
@@ -267,7 +266,6 @@ std::vector<float> conv2d_ref_xnnpack(
267266
static_cast<size_t>(H_in),
268267
static_cast<size_t>(W_in),
269268
&workspace_size,
270-
&workspace_alignment,
271269
&out_h,
272270
&out_w,
273271
/*threadpool=*/nullptr);
Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,8 +1,9 @@
11
load("@fbsource//xplat/executorch/build:runtime_wrapper.bzl", "runtime")
2+
load("@fbcode_macros//build_defs:build_file_migration.bzl", "fbcode_target")
23

34
oncall("executorch")
45

5-
runtime.python_library(
6+
fbcode_target(_kind = runtime.python_library,
67
name = "lib",
78
srcs = [
89
"__init__.py",

examples/models/gemma4_31b/BUCK

Lines changed: 40 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,40 @@
1+
load("@fbsource//xplat/executorch/build:runtime_wrapper.bzl", "runtime")
2+
load("@fbcode_macros//build_defs:build_file_migration.bzl", "fbcode_target")
3+
4+
oncall("executorch")
5+
6+
fbcode_target(_kind = runtime.python_library,
7+
name = "quant",
8+
srcs = [
9+
"cuda_packers.py",
10+
"quant/quantize.py",
11+
"quant/recipe.py",
12+
],
13+
base_module = "executorch.examples.models.gemma4_31b",
14+
visibility = ["PUBLIC"],
15+
deps = [
16+
"//caffe2:torch",
17+
"//executorch/backends/cuda:coalesced_int4_tensor",
18+
"//executorch/backends/cuda:dp4a_planar_int5_tensor",
19+
"//executorch/backends/cuda:dp4a_planar_int6_tensor",
20+
"//executorch/extension/llm/export:gguf",
21+
"//executorch/extension/llm/export:int4",
22+
"//executorch/extension/llm/export:quant",
23+
"//pytorch/ao:torchao",
24+
"fbsource//third-party/pypi/safetensors:safetensors",
25+
],
26+
)
27+
28+
fbcode_target(_kind = runtime.python_library,
29+
name = "quantize_and_save",
30+
srcs = ["quantize_and_save.py"],
31+
base_module = "executorch.examples.models.gemma4_31b",
32+
visibility = ["PUBLIC"],
33+
deps = [
34+
"//caffe2:torch",
35+
"//executorch/extension/llm/export:load",
36+
"//executorch/extension/llm/export:quant",
37+
"//pytorch/ao:torchao",
38+
"fbsource//third-party/pypi/safetensors:safetensors",
39+
],
40+
)

0 commit comments

Comments
 (0)