From 2f7b9f5bdaa2e90763f5272831db052cf88828e1 Mon Sep 17 00:00:00 2001
From: Ivan Kobzarev <ivankobzarev@fb.com>
Date: Thu, 3 Dec 2020 12:22:14 -0800
Subject: [PATCH] [vulkan][test] Not use non 1 dilation for conv2d

ghstack-source-id: 2b6f6068f388d5dcbbcf4bb7c0a3d1009098bd4c
Pull Request resolved: https://github.com/pytorch/pytorch/pull/48800
---
 aten/src/ATen/test/vulkan_api_test.cpp | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/aten/src/ATen/test/vulkan_api_test.cpp b/aten/src/ATen/test/vulkan_api_test.cpp
index 73b221c81b9d..cbd65fd9b68f 100644
--- 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;