@@ -188,7 +188,7 @@ def _get_analysis_config(
188188 def check_output (self , atol = 1e-5 ):
189189 '''
190190 Check whether calculating on CPU and GPU, enable TensorRT
191- or disable TensorRT, enable MKLDNN or disable MKLDNN
191+ or disable TensorRT, enable ONEDNN or disable ONEDNN
192192 are all the same.
193193 '''
194194 self .assertFalse (
@@ -203,7 +203,7 @@ def check_output_with_option(
203203 ):
204204 '''
205205 Check whether calculating on CPU and GPU, enable TensorRT
206- or disable TensorRT, enable MKLDNN or disable MKLDNN
206+ or disable TensorRT, enable ONEDNN or disable ONEDNN
207207 are all the same.
208208 '''
209209 place = base .CUDAPlace (0 ) if use_gpu else base .CPUPlace ()
@@ -295,7 +295,7 @@ def check_output_with_option(
295295
296296 self .assertTrue (
297297 len (paddle_outs ) == len (onednn_outputs ),
298- "The number of outputs is different between CPU and MKLDNN . " ,
298+ "The number of outputs is different between CPU and ONEDNN . " ,
299299 )
300300
301301 if self .enable_onednn_bfloat16 :
@@ -306,7 +306,7 @@ def check_output_with_option(
306306 onednn_output ,
307307 rtol = 1e-05 ,
308308 atol = atol ,
309- err_msg = 'Output has diff between CPU and MKLDNN . ' ,
309+ err_msg = 'Output has diff between CPU and ONEDNN . ' ,
310310 )
311311
312312 class TensorRTParam :
0 commit comments