rayorch 0.0.1__py3-none-any.whl
This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
- rayorch/__init__.py +57 -0
- rayorch/dag_new_pipeline.py +795 -0
- rayorch/dag_pipeline.py +372 -0
- rayorch/dispatch_mode.py +315 -0
- rayorch/env_registry.py +380 -0
- rayorch/nvtx_profiler.py +29 -0
- rayorch/overlapped_pipeline.py +96 -0
- rayorch/ray_module.py +246 -0
- rayorch/version.py +23 -0
- rayorch-0.0.1.dist-info/METADATA +82 -0
- rayorch-0.0.1.dist-info/RECORD +14 -0
- rayorch-0.0.1.dist-info/WHEEL +5 -0
- rayorch-0.0.1.dist-info/licenses/LICENSE +201 -0
- rayorch-0.0.1.dist-info/top_level.txt +1 -0
rayorch/__init__.py
ADDED
|
@@ -0,0 +1,57 @@
|
|
|
1
|
+
|
|
2
|
+
from .version import __version__, version_info
|
|
3
|
+
from .dispatch_mode import (
|
|
4
|
+
DispatchMode,
|
|
5
|
+
get_predefined_dispatch_fn,
|
|
6
|
+
Dispatch
|
|
7
|
+
)
|
|
8
|
+
from .ray_module import RayModule
|
|
9
|
+
|
|
10
|
+
RayModuleFuture = RayModule.RayModuleFuture
|
|
11
|
+
from .dag_pipeline import (
|
|
12
|
+
DagNode,
|
|
13
|
+
DagPipeline,
|
|
14
|
+
DagPipelineExecutor,
|
|
15
|
+
PipeRef,
|
|
16
|
+
PipelineExecutor,
|
|
17
|
+
)
|
|
18
|
+
from .dag_new_pipeline import (
|
|
19
|
+
Pipeline,
|
|
20
|
+
DagPipeline as DagNewPipeline,
|
|
21
|
+
PipeRef as DagNewPipeRef,
|
|
22
|
+
Executor,
|
|
23
|
+
SequentialExecutor,
|
|
24
|
+
DagExecutor,
|
|
25
|
+
)
|
|
26
|
+
from .overlapped_pipeline import OverlappedPipeline
|
|
27
|
+
from .env_registry import EnvRegistry
|
|
28
|
+
|
|
29
|
+
|
|
30
|
+
|
|
31
|
+
__all__ = [
|
|
32
|
+
# General
|
|
33
|
+
'__version__',
|
|
34
|
+
'version_info',
|
|
35
|
+
# Dispatch Mode Related
|
|
36
|
+
'DispatchMode',
|
|
37
|
+
'get_predefined_dispatch_fn',
|
|
38
|
+
'Dispatch',
|
|
39
|
+
# Main RayModule
|
|
40
|
+
'RayModule',
|
|
41
|
+
'RayModuleFuture',
|
|
42
|
+
'DagNode',
|
|
43
|
+
'DagPipeline',
|
|
44
|
+
'DagPipelineExecutor',
|
|
45
|
+
'Pipeline',
|
|
46
|
+
'DagNewPipeline',
|
|
47
|
+
'DagNewPipeRef',
|
|
48
|
+
'Executor',
|
|
49
|
+
'SequentialExecutor',
|
|
50
|
+
'DagExecutor',
|
|
51
|
+
'OverlappedPipeline',
|
|
52
|
+
'PipelineExecutor',
|
|
53
|
+
'PipeRef',
|
|
54
|
+
]
|
|
55
|
+
|
|
56
|
+
def hello():
|
|
57
|
+
return "Hello from RayOrch!"
|