Skip to content

Commit 23eba43

Browse files
co63ocSigureMo
andcommitted
[SOT] Support builtin dispatch for is_compiled_with_onednn (PaddlePaddle#75747)
--------- Co-authored-by: Nyakku Shigure <sigure.qaq@gmail.com>
1 parent 3cd100a commit 23eba43

2 files changed

Lines changed: 6 additions & 1 deletion

File tree

python/paddle/jit/sot/utils/paddle_api_config.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -149,6 +149,7 @@ def is_directly_run_api(api):
149149
paddle.base.libpaddle.is_compiled_with_ipu,
150150
paddle.base.libpaddle.is_compiled_with_xpu,
151151
paddle.base.libpaddle.is_compiled_with_mkldnn,
152+
paddle.base.libpaddle.is_compiled_with_onednn,
152153
paddle.base.libpaddle.is_compiled_with_nccl,
153154
paddle.base.libpaddle.is_compiled_with_mpi,
154155
paddle.base.libpaddle.is_compiled_with_mpi_aware,

test/sot/test_builtin_dispatch.py

Lines changed: 5 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -458,7 +458,10 @@ def test_native_code_function():
458458
res5 = paddle.base.libpaddle.is_compiled_with_custom_device("npu")
459459
res6 = paddle.base.libpaddle.is_compiled_with_ipu()
460460
res7 = paddle.base.libpaddle.is_compiled_with_xpu()
461-
res8 = paddle.base.libpaddle.is_compiled_with_mkldnn()
461+
res8_deprecated = (
462+
paddle.base.libpaddle.is_compiled_with_mkldnn()
463+
) # Paddle 3.3 deprecated
464+
res8 = paddle.base.libpaddle.is_compiled_with_onednn()
462465
res9 = paddle.base.libpaddle.is_compiled_with_nccl()
463466
res10 = paddle.base.libpaddle.is_compiled_with_mpi()
464467
res11 = paddle.base.libpaddle.is_compiled_with_mpi_aware()
@@ -474,6 +477,7 @@ def test_native_code_function():
474477
res5,
475478
res6,
476479
res7,
480+
res8_deprecated,
477481
res8,
478482
res9,
479483
res10,

0 commit comments

Comments
 (0)