File tree Expand file tree Collapse file tree 2 files changed +12
-0
lines changed Expand file tree Collapse file tree 2 files changed +12
-0
lines changed Original file line number Diff line number Diff line change 45
45
from monarch .common .opaque_ref import OpaqueRef
46
46
from monarch .common .pipe import create_pipe , Pipe , remote_generator
47
47
from monarch .common .remote import remote
48
+ from monarch .common .selection import Selection
48
49
from monarch .common .shape import NDSlice , Shape
49
50
from monarch .common .stream import get_active_stream , Stream
50
51
from monarch .common .tensor import reduce , reduce_ , Tensor
76
77
"RemoteException" : ("monarch.common.invocation" , "RemoteException" ),
77
78
"Shape" : ("monarch.common.shape" , "Shape" ),
78
79
"NDSlice" : ("monarch.common.shape" , "NDSlice" ),
80
+ "Selection" : ("monarch.common.selection" , "Selection" ),
79
81
"OpaqueRef" : ("monarch.common.opaque_ref" , "OpaqueRef" ),
80
82
"create_pipe" : ("monarch.common.pipe" , "create_pipe" ),
81
83
"Pipe" : ("monarch.common.pipe" , "Pipe" ),
@@ -143,6 +145,7 @@ def __getattr__(name):
143
145
"Future" ,
144
146
"RemoteException" ,
145
147
"Shape" ,
148
+ "Selection" ,
146
149
"NDSlice" ,
147
150
"OpaqueRef" ,
148
151
"create_pipe" ,
Original file line number Diff line number Diff line change
1
+ # Copyright (c) Meta Platforms, Inc. and affiliates.
2
+ # All rights reserved.
3
+ #
4
+ # This source code is licensed under the BSD-style license found in the
5
+ # LICENSE file in the root directory of this source tree.
6
+
7
+ from monarch ._rust_bindings .monarch_hyperactor .selection import Selection
8
+
9
+ __all__ = ["Selection" ]
You can’t perform that action at this time.
0 commit comments