mirror of
https://github.com/saymrwulf/pytorch.git
synced 2026-05-14 20:57:59 +00:00
e363a8a222
This reverts commit b948b1ad7a.
Reverted https://github.com/pytorch/pytorch/pull/126721 on behalf of https://github.com/clee2000 due to The test_public_bindings failure is real, you just got unlucky since it was also broken on trunk for a different reason ([comment](https://github.com/pytorch/pytorch/pull/126721#issuecomment-2121725408))
32 lines
631 B
Python
32 lines
631 B
Python
# Copyright (c) Meta Platforms, Inc. and affiliates
|
|
from ._IR import (
|
|
annotate_split_points,
|
|
ArgsChunkSpec,
|
|
KwargsChunkSpec,
|
|
Pipe,
|
|
pipe_split,
|
|
pipeline,
|
|
SplitPoint,
|
|
)
|
|
from ._PipelineStage import PipelineStage
|
|
from .PipelineSchedule import (
|
|
Schedule1F1B,
|
|
ScheduleGPipe,
|
|
ScheduleInterleaved1F1B,
|
|
ScheduleLoopedBFS,
|
|
)
|
|
|
|
__all__ = [
|
|
"Pipe",
|
|
"pipe_split",
|
|
"SplitPoint",
|
|
"annotate_split_points",
|
|
"pipeline",
|
|
"ArgsChunkSpec",
|
|
"KwargsChunkSpec",
|
|
"PipelineStage",
|
|
"Schedule1F1B",
|
|
"ScheduleGPipe",
|
|
"ScheduleInterleaved1F1B",
|
|
"ScheduleLoopedBFS",
|
|
]
|