Skip to content

Commit 762ff6a

Browse files
zonglinpengfacebook-github-bot
authored andcommitted
link M2 g3 operator kernels to jarvis and BUCK dep (#7529)
Summary: - Apply buck fix to Cadence's diff. - resolved duplicate work internally - fixed naming of mean link to jarvis - sub (fp32, int32), - div (fp32, int32), - exp (fp32), - permute (8-bit, 16-bit and 32-bit width) - slice (8-bit, 16-bit and 32-bit width). - mean (fp32) Reviewed By: hsharma35 Differential Revision: D67875050
1 parent 0c4053e commit 762ff6a

File tree

2 files changed

+7
-2
lines changed

2 files changed

+7
-2
lines changed

backends/cadence/fusion_g3/operators/op_mean.cpp

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -59,7 +59,7 @@ int prepare_data(
5959
return num_axis_dims;
6060
}
6161

62-
Tensor& mean_dim_out(
62+
Tensor& mean_out(
6363
KernelRuntimeContext& ctx,
6464
const Tensor& in,
6565
optional<ArrayRef<int64_t>> dim_list,
@@ -199,4 +199,4 @@ Tensor& mean_dim_out(
199199
} // namespace native
200200
} // namespace G3
201201
} // namespace impl
202-
} // namespace cadence
202+
} // namespace cadence

backends/cadence/fusion_g3/operators/targets.bzl

Lines changed: 5 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -36,9 +36,14 @@ OPERATORS = [
3636
"add",
3737
"cat",
3838
"dequantize",
39+
"div",
40+
"exp",
41+
"mean",
3942
"mul",
4043
"native_layer_norm",
44+
"permute_copy",
4145
"quantize",
46+
"slice_copy",
4247
"softmax",
4348
"sub",
4449
"div",

0 commit comments

Comments
 (0)