MCPcopy Create free account
hub / github.com/pytorch/pytorch / temp_workspace

Function temp_workspace

caffe2/python/hypothesis_test_util.py:334–339  ·  view source on GitHub ↗
(name=b"temp_ws")

Source from the content-addressed store, hash-verified

332
333@contextlib.contextmanager
334def temp_workspace(name=b"temp_ws"):
335 old_ws_name = workspace.CurrentWorkspace()
336 workspace.SwitchWorkspace(name, True)
337 yield
338 workspace.ResetWorkspace()
339 workspace.SwitchWorkspace(old_ws_name)
340
341
342def runOpBenchmark(

Callers 5

runOpBenchmarkFunction · 0.85
runOpOnInputFunction · 0.85
assertReferenceChecksMethod · 0.85
assertRunOpRaisesMethod · 0.85

Calls

no outgoing calls

Tested by

no test coverage detected

Used in the wild real call sites across dependent graphs

searching dependent graphs…