From 1358e03c3f204105389efc6c894272a3ec987aef Mon Sep 17 00:00:00 2001 From: Jorge Pineda Date: Sat, 1 Jun 2024 00:45:25 -0700 Subject: [PATCH] Fix compute_graph_op_tests (#3796) Summary: Pull Request resolved: https://github.com/pytorch/executorch/pull/3796 TSIA Reviewed By: copyrightly Differential Revision: D58047889 fbshipit-source-id: 78fa262abdd096b473b7d280370d46a40908473d --- backends/vulkan/test/op_tests/cases.py | 1 + 1 file changed, 1 insertion(+) diff --git a/backends/vulkan/test/op_tests/cases.py b/backends/vulkan/test/op_tests/cases.py index e558966cc2..7eb9f1936d 100644 --- a/backends/vulkan/test/op_tests/cases.py +++ b/backends/vulkan/test/op_tests/cases.py @@ -166,6 +166,7 @@ def get_avg_pool2d_inputs(): ), ] test_suite = VkTestSuite([tuple(tc) for tc in test_cases]) + test_suite.dtypes = ["at::kFloat"] return test_suite