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