Skip to content

Commit 7a5b348

Browse files
author
morelos
committed
Update on "[ET-VK] double, short, and uint16 dtype runtime support"
Creating support for double, short, and uint16 for quantization ops. Registering the short keyword since theres already support. Also changing the cpu implementation to support half Differential Revision: [D75959063](https://our.internmc.facebook.com/intern/diff/D75959063/) [ghstack-poisoned]
2 parents db8f9c0 + 25063da commit 7a5b348

File tree

3 files changed

+3
-3
lines changed

3 files changed

+3
-3
lines changed

backends/vulkan/runtime/graph/ops/glsl/arange.yaml

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -7,7 +7,7 @@
77
arange:
88
parameter_names_with_default_values:
99
NDIM: 3
10-
DTYPE: int
10+
DTYPE: int32
1111
STORAGE: texture3d
1212
PACKING: C_packed
1313
generate_variant_forall:

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

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -17,7 +17,7 @@ unary_op:
1717
OPERATOR: clamp(X, A, B)
1818
- NAME: clamp_int32
1919
OPERATOR: clamp(X, A, B)
20-
DTYPE: int
20+
DTYPE: int32
2121
- NAME: cos
2222
OPERATOR: cos(X)
2323
- NAME: exp

backends/vulkan/tools/gpuinfo/glsl/warp_size.yaml

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -6,7 +6,7 @@
66

77
warp_size:
88
parameter_names_with_default_values:
9-
DTYPE: int
9+
DTYPE: int32
1010
STORAGE: buffer
1111
generate_variant_forall:
1212
METHOD:

0 commit comments

Comments
 (0)