enable_compat
- paddle. enable_compat ( *, scope=None, blocked_modules=None, backend='torch', silent=False ) [源代码]
通过将 TorchProxyMetaFinder 添加到 sys.meta_path 来启用 PyTorch 兼容代理。 这允许导入实际上是 PaddlePaddle 代理的 torch 模块。
参数
scope (str|Iterable[str]，可选) - 指定启用 PyTorch 兼容代理的模块或模块列表。如果为
None，则全局启用 PyTorch 兼容代理。默认为None。blocked_modules (str|Iterable[str]，可选) - 指定从 PyTorch 兼容代理中排除的模块或模块列表。默认为
None。backend (str，可选) - 要启用兼容性的后端。目前仅支持
"torch"。默认为"torch"。silent (bool，可选) - 如果为
True，则抑制有关范围更改的警告。默认为False。
返回
None
代码示例 1
>>> import paddle
>>> paddle.enable_compat() # Enable torch compat globally
>>> import torch # This will import paddle as torch
>>> assert torch.sin is paddle.sin
>>> paddle.disable_compat() # Disable torch compat
代码示例 2
>>> import paddle
>>> paddle.enable_compat(scope={"triton"}) # Enable torch compat for 'triton' module only
>>> import triton # All `import torch` inside `triton` will proxy to paddle
>>> try:
... import torch # This will raise ModuleNotFoundError
... except ModuleNotFoundError:
... print("PyTorch compat is not enabled globally.")
>>> paddle.disable_compat() # Disable torch compat