MCPcopy Index your code
hub / github.com/pytorch/pytorch / local_exit

Method local_exit

caffe2/python/net_builder.py:445–453  ·  view source on GitHub ↗

Similar to `task_exit`, but executes at TaskGroup's exit instead, after all tasks of the group finished execution. This will run only once on each node.

(self, name=None)

Source from the content-addressed store, hash-verified

443 return setup
444
445 def local_exit(self, name=None):
446 """
447 Similar to `task_exit`, but executes at TaskGroup's exit instead,
448 after all tasks of the group finished execution.
449 This will run only once on each node.
450 """
451 setup = _SetupBuilder(_SetupBuilder.EXIT, name)
452 self.net().add_attribute(TaskGroup.LOCAL_SETUP, setup)
453 return setup
454
455 def task_reporter(self, interval_ms=1000, name=None):
456 """

Callers

nothing calls this directly

Calls 3

netMethod · 0.95
_SetupBuilderClass · 0.85
add_attributeMethod · 0.80

Tested by

no test coverage detected