no EIGEN engine for DeformConv (#16785)
authorJongsoo Park <jongsoo@fb.com>
Wed, 6 Feb 2019 19:52:23 +0000 (11:52 -0800)
committerFacebook Github Bot <facebook-github-bot@users.noreply.github.com>
Wed, 6 Feb 2019 19:59:31 +0000 (11:59 -0800)
Summary:
Pull Request resolved: https://github.com/pytorch/pytorch/pull/16785

There's no EIGEN engine implemented for DeformConv but unit test was checking it.

Reviewed By: BIT-silence

Differential Revision: D13967306

fbshipit-source-id: e29c19f59f5700fc0501c59f45d60443b87ffedc

caffe2/python/operator_test/deform_conv_test.py

index b42b3bd..31e4074 100644 (file)
@@ -430,7 +430,6 @@ class TestConvolution(hu.HypothesisTestCase):
         output_channels=st.integers(1, 3),
         batch_size=st.integers(1, 3),
         order=st.sampled_from(["NCHW"]),
-        engine=st.sampled_from(["", "EIGEN"]),
         shared_buffer=st.booleans(),
         use_bias=st.booleans(),
         deformable_group=st.integers(1, 3),
@@ -448,7 +447,6 @@ class TestConvolution(hu.HypothesisTestCase):
         output_channels,
         batch_size,
         order,
-        engine,
         shared_buffer,
         use_bias,
         deformable_group,
@@ -467,7 +465,6 @@ class TestConvolution(hu.HypothesisTestCase):
             pad_r=pad_w,
             kernel=kernel,
             order=order,
-            engine=engine,
             shared_buffer=int(shared_buffer),
             deformable_group=deformable_group,
         )