fix

Author: co63oc

test/deprecated/quantization/test_quantization_onednn_pass_deprecated.py

@@ -19,7 +19,7 @@
 import numpy as np
 
 import paddle
-from paddle.base.framework import IrGraph, _current_expected_place
+from paddle.base.framework import IrGraph
 from paddle.framework import core
 from paddle.static.quantization import (
     QuantInt8OnednnPass,
@@ -60,10 +60,6 @@ def conv_net(img, label):
     return avg_loss
 
 
-@unittest.skipIf(
-    not isinstance(_current_expected_place(), core.CPUPlace),
-    "OneDNN supports only CPU for now",
-)
 class TestONEDNNTransformBasedFreezePass(unittest.TestCase):
     def setUp(self):
         self.quantizable_op_and_inputs = {