add a note on dilation in x86

apache · Feb 29, 2020 · e410221 · e410221
1 parent 8336237
commit e410221
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/tests/python/frontend/pytorch/test_forward.py b/tests/python/frontend/pytorch/test_forward.py
@@ -797,8 +797,10 @@ def forward(self, inp):
     inp = [torch.rand((1, 3, 300, 300), dtype=torch.float)]
 
     for model in [fcn, deeplab]:
+        # depthwise + dilated covolution not supported on x86
+        # see https://github.com/apache/incubator-tvm/issues/4962
         verify_model(SegmentationModelWrapper(model.eval()), inp,
-                     ctx_list=[("cuda", tvm.gpu(0))])  # dilated covolution not supported on x86
+                     ctx_list=[("cuda", tvm.gpu(0))])
 
 
 if __name__ == "__main__":