@@ -196,7 +196,7 @@ def setUp(self):
196196 self .use_cudnn = False
197197 self .exhaustive_search = False
198198 self .use_cuda = False
199- self .use_mkldnn = False
199+ self .use_onednn = False
200200 self .fuse_relu_before_depthwise_conv = False
201201 self .data_format = "AnyLayout"
202202 self .init_kernel_type ()
@@ -241,7 +241,7 @@ def setUp(self):
241241 'groups' : self .groups ,
242242 'dilations' : self .dilations ,
243243 'use_cudnn' : self .use_cudnn ,
244- 'use_mkldnn' : self .use_mkldnn ,
244+ 'use_mkldnn' : self .use_onednn ,
245245 'data_format' : self .data_format ,
246246 'fuse_relu_before_depthwise_conv' : self .fuse_relu_before_depthwise_conv ,
247247 'exhaustive_search' : self .exhaustive_search ,
@@ -350,7 +350,7 @@ def setUp(self):
350350 self .use_cudnn = False
351351 self .exhaustive_search = False
352352 self .use_cuda = False
353- self .use_mkldnn = False
353+ self .use_onednn = False
354354 self .fuse_relu_before_depthwise_conv = False
355355 self .init_kernel_type ()
356356 self .init_group ()
@@ -402,7 +402,7 @@ def setUp(self):
402402 'groups' : self .groups ,
403403 'dilations' : self .dilations ,
404404 'use_cudnn' : self .use_cudnn ,
405- 'use_mkldnn' : self .use_mkldnn ,
405+ 'use_mkldnn' : self .use_onednn ,
406406 'data_format' : self .data_format ,
407407 'fuse_relu_before_depthwise_conv' : self .fuse_relu_before_depthwise_conv ,
408408 'exhaustive_search' : self .exhaustive_search ,
0 commit comments