Fix unbound local error for `return_val` (#7395)
......
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
Signed-off-by: Hollow Man <hollowman@opensuse.org>
Co-authored-by: Masahiro Tanaka <81312776+tohtana@users.noreply.github.com>
