Fix unbound local error for return_val
#7395
Merged
Add this suggestion to a batch that can be applied as a single commit.
This suggestion is invalid because no changes were made to the code.
Suggestions cannot be applied while the pull request is closed.
Suggestions cannot be applied while viewing a subset of changes.
Only one suggestion per line can be applied in a batch.
Add this suggestion to a batch that can be applied as a single commit.
Applying suggestions on deleted lines is not supported.
You must change the existing code in this line in order to create a valid suggestion.
Outdated suggestions cannot be applied.
This suggestion has been applied or marked resolved.
Suggestions cannot be applied from pending reviews.
Suggestions cannot be applied on multi-line comments.
Suggestions cannot be applied while the pull request is queued to merge.
Suggestion cannot be applied right now. Please check back later.
......
File "torch/_dynamo/backends/common.py", line 72, in _wrapped_bw_compiler
return disable(disable(bw_compiler_fn)(*args, **kwargs))
File "torch/_dynamo/eval_frame.py", line 838, in _fn
return fn(*args, **kwargs)
File "deepspeed/compile/inductor.py", line 27, in wrapped_compiler
mod_graph = dc_compiler(gm, fake_inputs)
File "deepspeed/compile/backend.py", line 330, in make_bw_graph
run_opt_passes(
File "deepspeed/compile/backend.py", line 206, in run_opt_passes
mem_prof.run(*create_inputs_fn())
File "deepspeed/compile/profilers/graph_profile.py", line 261, in run
return return_val
UnboundLocalError: local variable 'return_val' referenced before assignment