[vulkan][test] Not use non 1 dilation for conv2d

ghstack-source-id: 2b6f6068f388d5dcbbcf4bb7c0a3d1009098bd4c Pull Request resolved: #48800
pytorch · Dec 3, 2020 · 2f7b9f5 · 2f7b9f5
1 parent 1eed54d
commit 2f7b9f5
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/aten/src/ATen/test/vulkan_api_test.cpp b/aten/src/ATen/test/vulkan_api_test.cpp
@@ -269,7 +269,8 @@ TEST(VulkanAPITest, conv2d) {
   constexpr int64_t groups = 1;
   constexpr std::array<int64_t, 2u> stride{1, 2};
   constexpr std::array<int64_t, 2u> padding{3, 0};
-  constexpr std::array<int64_t, 2u> dilation{1, 3};
+  //TODO: Support conv2d with dilation != 1
+  constexpr std::array<int64_t, 2u> dilation{1, 1};
 
   constexpr struct {
     uint32_t batches;