utils

Classes

`ActivityContext`
`DynamoDisable`
`DynamoSkip`
`ModuleRef`
`MyFakeTensor`

Functions

`distributed_isend_obj`
`distributed_recv_obj`
`distributed_send_obj`
`fake_tensor`
`fake_tensor_like`
`fake_tensors`
`has_fake_tensor`
`is_fake_tensor`
`is_submodule_of`
`is_submodule_or_same`
`real_tensors`

class ActivityContext

Bases: Generic[T]

__init__(max_depth=None, no_duplicates=False, reversed=False)

Parameters:: max_depth (int | None)

get_active()

Return type:: T | None

is_active()

Return type:: bool

exception ActivityContextDuplicateException: Bases: Exception

exception ActivityContextMaxDepthException: Bases: Exception

class DynamoDisable

Bases: Protocol

__init__(*args, **kwargs)

class DynamoSkip

Bases: Protocol

__init__(*args, **kwargs)

class ModuleRef

Bases: Generic[TModule]

__init__(module)

Parameters:: module (TModule)

class MyFakeTensor

Bases: Tensor

__init__(*args, **kwargs)

static __new__(cls, elem, device)

Return type:: MyFakeTensor

classmethod create(data)

Parameters:: data (Tensor)
Return type:: MyFakeTensor

property device

distributed_isend_obj(obj, dst=0, group=None)

Parameters:

obj (Any)
dst (int)
group (ProcessGroup | None)

Return type:

list[Work | None]

distributed_recv_obj(src=None, group=None)

Parameters:

src (int | None)
group (ProcessGroup | None)

Return type:

Any

distributed_send_obj(obj, dst=0, group=None)

Parameters:

obj (Any)
dst (int)
group (ProcessGroup | None)

fake_tensor(t: Tensor, *, dtype: dtype | None = None, use_meta=False) → Tensor
fake_tensor(size: Sequence[int] | torch.Size, *, dtype: dtype | None = None, use_meta=False) → Tensor
fake_tensor(*args: int, dtype: dtype | None = None, use_meta=False) → Tensor

fake_tensor_like(t, use_meta=False)

Parameters:: t (Tensor)
Return type:: Tensor

fake_tensors(value, use_meta=False)

Parameters:: value (T)
Return type:: T

has_fake_tensor(v)

Parameters:: v (Any)
Return type:: bool

is_fake_tensor(t)

Parameters:: t (Any)
Return type:: bool

is_submodule_of(module_name, other_module_name)

Parameters:

module_name (str)
other_module_name (str)

Return type:

bool

is_submodule_or_same(module_name, other_module_name)

Parameters:

module_name (str)
other_module_name (str)

Return type:

bool

real_tensors(value)

Parameters:: value (Any)
Return type:: Any