remove profile

Author: lanluo-nvidia

tools/perf/Flux/flux_perf.py

@@ -9,24 +9,6 @@
 from flux_demo import compile_model
 
 
-def profile(pipe, prompt, inference_step, batch_size=1):
-    print(f"Running torch profiler with {inference_step=} {batch_size=}")
-    with torch.profiler.profile(
-        activities=[torch.profiler.ProfilerActivity.CUDA],
-        record_shapes=True,
-        profile_memory=True,
-        with_stack=True,
-    ) as prof:
-        with torch.profiler.record_function("model_inference"):
-            pipe(
-                prompt,
-                output_type="pil",
-                num_inference_steps=inference_step,
-                num_images_per_prompt=batch_size,
-            ).images
-    print(prof.key_averages().table(sort_by="cuda_time_total", row_limit=100))
-
-
 def benchmark(pipe, prompt, inference_step, batch_size=1, iterations=1):
     print(f"Running warmup with {batch_size=} {inference_step=} iterations=10")
     # warmup
@@ -67,7 +49,6 @@ def main(args):
     pipe, backbone, trt_gm = compile_model(args)
 
     benchmark(pipe, ["Test"], 20, batch_size=args.max_batch_size, iterations=3)
-    # profile(pipe, ["enchanted winter forest, soft diffuse light on a snow-filled day, serene nature scene, the forest is illuminated by the snow"], 20, batch_size=args.max_batch_size)
 
 
 if __name__ == "__main__":