提交 4f99f000 authored 作者: affanv14's avatar affanv14

add is_dnn attribute to gemm tests

上级 c36a5ad8
...@@ -228,3 +228,4 @@ class TestGroupGpuCorr2d(Grouped_conv_noOptim): ...@@ -228,3 +228,4 @@ class TestGroupGpuCorr2d(Grouped_conv_noOptim):
conv2d_gradw = staticmethod(GpuCorrMM_gradWeights) conv2d_gradw = staticmethod(GpuCorrMM_gradWeights)
conv2d_gradi = staticmethod(GpuCorrMM_gradInputs) conv2d_gradi = staticmethod(GpuCorrMM_gradInputs)
flip_filter = True flip_filter = True
is_dnn = False
...@@ -426,6 +426,7 @@ class TestGroupCorr2d(Grouped_conv_noOptim): ...@@ -426,6 +426,7 @@ class TestGroupCorr2d(Grouped_conv_noOptim):
conv2d_gradw = staticmethod(corr.CorrMM_gradWeights) conv2d_gradw = staticmethod(corr.CorrMM_gradWeights)
conv2d_gradi = staticmethod(corr.CorrMM_gradInputs) conv2d_gradi = staticmethod(corr.CorrMM_gradInputs)
flip_filter = True flip_filter = True
is_dnn = False
def test_graph(self): def test_graph(self):
# define common values first # define common values first
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论