Fix compute_graph_op_tests (pytorch#3796)

Summary: Pull Request resolved: pytorch#3796 TSIA Reviewed By: copyrightly Differential Revision: D58047889 fbshipit-source-id: 78fa262abdd096b473b7d280370d46a40908473d
mybigday · Jun 1, 2024 · 1358e03 · 1358e03
1 parent fa3bf82
commit 1358e03
Showing 1 changed file with 1 addition and 0 deletions.
diff --git 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