diff --git a/pytorch/optimized.py b/pytorch/optimized.py index 8c70785..ec0aae4 100644 --- a/pytorch/optimized.py +++ b/pytorch/optimized.py @@ -35,7 +35,6 @@ def main(): # CUDA Graphs to reduce launch overhead when shapes are static graph = None - s_out = torch.zeros((), dtype=torch.float32, device=device) if device.type == "cuda": stream = torch.cuda.Stream() torch.cuda.synchronize()