tensorplay.distributed.fsdp.register_fsdp_forward_method

tensorplay.distributed.fsdp.register_fsdp_forward_method(module: Any, method_name: str) → None[source]