Fix profiling will crash the server when using num_steps (#6586)
This commit is contained in:
@@ -2203,6 +2203,7 @@ class Scheduler(
|
|||||||
self.torch_profiler = None
|
self.torch_profiler = None
|
||||||
self.torch_profiler_output_dir = None
|
self.torch_profiler_output_dir = None
|
||||||
self.profiler_activities = None
|
self.profiler_activities = None
|
||||||
|
self.profiler_target_forward_ct = None
|
||||||
|
|
||||||
return ProfileReqOutput(success=True, message="Succeeded")
|
return ProfileReqOutput(success=True, message="Succeeded")
|
||||||
|
|
||||||
|
|||||||
Reference in New Issue
Block a user