From ae0776fed7432562f350d4af61c5e6cd559a5308 Mon Sep 17 00:00:00 2001 From: Jorge Pineda Date: Wed, 13 Mar 2024 16:43:51 -0700 Subject: [PATCH] [ET-VK][EZ] Include alpha to fix aten.sub.Tensor Missed this in https://github.com/pytorch/executorch/pull/2366 Differential Revision: [D54880024](https://our.internmc.facebook.com/intern/diff/D54880024/) [ghstack-poisoned] --- backends/vulkan/runtime/graph/ops/glsl/all_shaders.yaml | 2 +- backends/vulkan/test/test_vulkan_delegate.py | 8 ++++---- 2 files changed, 5 insertions(+), 5 deletions(-) diff --git a/backends/vulkan/runtime/graph/ops/glsl/all_shaders.yaml b/backends/vulkan/runtime/graph/ops/glsl/all_shaders.yaml index f954528ee7e..a1abc6a7454 100644 --- a/backends/vulkan/runtime/graph/ops/glsl/all_shaders.yaml +++ b/backends/vulkan/runtime/graph/ops/glsl/all_shaders.yaml @@ -19,7 +19,7 @@ binary_op: shader_variants: - NAME: binary_add - NAME: binary_sub - OPERATOR: X - Y + OPERATOR: X - A * Y - NAME: binary_mul OPERATOR: X * Y - NAME: binary_div diff --git a/backends/vulkan/test/test_vulkan_delegate.py b/backends/vulkan/test/test_vulkan_delegate.py index 7def98df711..d18ca74588f 100644 --- a/backends/vulkan/test/test_vulkan_delegate.py +++ b/backends/vulkan/test/test_vulkan_delegate.py @@ -124,8 +124,8 @@ def __init__(self): self.weight = torch.rand(size=(2, 3), dtype=torch.float32) def forward(self, x, y): - z = x + y - z = z + x + z = torch.add(x, y, alpha=2) + z = torch.add(x, y, alpha=3.14) z = z + x z = z + self.weight return z @@ -144,8 +144,8 @@ def __init__(self): super().__init__() def forward(self, x, y): - z = x - y - z = z - x + z = torch.sub(x, y, alpha=2) + z = torch.sub(z, x, alpha=3.14) z = z - x return z