Skip to content

Commit d2f29f7

Browse files
author
Nyako Shigure
authored
fix deprecated paddle compat api usage (#11)
1 parent 5336436 commit d2f29f7

5 files changed

Lines changed: 5 additions & 5 deletions

File tree

README.md

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -35,7 +35,7 @@ Kernel Library for LLM Serving ❤️ PaddlePadddle
3535
>
3636
> ```python
3737
> import paddle
38-
> paddle.compat.enable_torch_proxy(scope={"flashinfer"}) # Enable torch proxy before importing flashinfer
38+
> paddle.enable_compat(scope={"flashinfer"}) # Enable torch proxy before importing flashinfer
3939
> import flashinfer
4040
> # use flashinfer
4141
> ```

tests/attention/test_attention_sink_blackwell.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -16,7 +16,7 @@
1616

1717
import paddle
1818

19-
paddle.compat.enable_torch_proxy()
19+
paddle.enable_compat()
2020
import einops
2121
import pytest
2222
import torch

tests/comm/test_trtllm_allreduce_fusion_paddle.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -73,7 +73,7 @@ def kernel(workspace_tensor, rank, world_size):
7373

7474
def _run_simple_worker(world_size, rank, distributed_init_port):
7575
# Create workspace
76-
# paddle.compat.enable_torch_proxy()
76+
# paddle.enable_compat()
7777
# Set all required environment variables
7878
os.environ["FLAGS_SELECTED_GPUS"] = str(rank) # Key: set GPU ID
7979
os.environ["PADDLE_TRAINER_ID"] = str(rank)

tests/conftest.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -6,7 +6,7 @@
66

77
import paddle
88

9-
paddle.compat.enable_torch_proxy()
9+
paddle.enable_compat()
1010
import pytest
1111
import torch
1212
# from torch.torch_version import TorchVersion

tests/moe/test_trtllm_gen_fused_moe.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -16,7 +16,7 @@
1616

1717
import paddle
1818

19-
paddle.compat.enable_torch_proxy()
19+
paddle.enable_compat()
2020
import functools
2121
from typing import Tuple
2222
from abc import ABC, abstractmethod

0 commit comments

Comments
 (0)