flex_model.distributed.NullMemoryOffloadStrategy

class flex_model.distributed.NullMemoryOffloadStrategy(offload_fn)

__init__(offload_fn)

Methods

__init__(offload_fn)
execute(tensor)
initialize(name, output_ptr)