Files
pytorch/torch/_streambase.py
Yu, Guangye d29094888b Use torch.Stream&torch.Event for Dynamo capature (#134850)
# Motivation
This PR aims to solve the multiple Inheritance problem.

Pull Request resolved: https://github.com/pytorch/pytorch/pull/134850
Approved by: https://github.com/yf225, https://github.com/EikanWang
2024-10-02 14:15:33 +00:00

21 lines
435 B
Python

from typing_extensions import deprecated
import torch
# Preserved only for BC reasons
@deprecated(
"`torch._streambase._StreamBase` is deprecated. Please use `torch.Stream` instead.",
category=FutureWarning,
)
class _StreamBase(torch.Stream):
pass
@deprecated(
"`torch._streambase._EventBase` is deprecated. Please use `torch.Event` instead.",
category=FutureWarning,
)
class _EventBase(torch.Event):
pass