Skip to content

Commit 4dab143

Browse files
committed
[ET-VK][ops] Add log10 support
Add `aten.log10.default` to the existing Vulkan unary shader family and cover float behavior across buffer and texture storage. Authored with Codex. Differential Revision: [D114382686](https://our.internmc.facebook.com/intern/diff/D114382686/) ghstack-source-id: 409188434 Pull-Request: pytorch#21520
1 parent d95b409 commit 4dab143

3 files changed

Lines changed: 5 additions & 0 deletions

File tree

‎backends/vulkan/runtime/graph/ops/glsl/unary_op.yaml‎

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -44,6 +44,8 @@ unary_op:
4444
OPERATOR: hardsigmoid(X)
4545
- NAME: leaky_relu
4646
OPERATOR: leaky_relu(X, A)
47+
- NAME: log10
48+
OPERATOR: log(X) * 0.4342944819
4749
- NAME: round
4850
OPERATOR: round(X)
4951
- NAME: bitwise_not_uint8

‎backends/vulkan/runtime/graph/ops/impl/UnaryOp.cpp‎

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -157,6 +157,7 @@ DEFINE_HARDSHRINK_FN(hardshrink);
157157
DEFINE_ACTIVATION_FN(hardswish);
158158
DEFINE_ACTIVATION_FN(hardsigmoid);
159159
DEFINE_LEAKY_RELU_FN(leaky_relu);
160+
DEFINE_ACTIVATION_FN(log10);
160161
DEFINE_ACTIVATION_FN(round);
161162
DEFINE_ACTIVATION_FN(bitwise_not);
162163

@@ -179,6 +180,7 @@ REGISTER_OPERATORS {
179180
VK_REGISTER_OP(aten.hardswish.default, hardswish);
180181
VK_REGISTER_OP(aten.hardsigmoid.default, hardsigmoid);
181182
VK_REGISTER_OP(aten.leaky_relu.default, leaky_relu);
183+
VK_REGISTER_OP(aten.log10.default, log10);
182184
VK_REGISTER_OP(aten.round.default, round);
183185
VK_REGISTER_OP(aten.bitwise_not.default, bitwise_not);
184186
}

‎backends/vulkan/test/op_tests/cases.py‎

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1813,6 +1813,7 @@ def get_var_inputs():
18131813
"aten.hardswish.default",
18141814
"aten.hardsigmoid.default",
18151815
"aten.leaky_relu.default",
1816+
"aten.log10.default",
18161817
"aten.round.default",
18171818
"aten.tan.default",
18181819
"aten.relu6.default",

0 commit comments

Comments
 (0)