This commit is contained in:
sherlock 2025-09-06 17:48:51 +05:30
parent baa9bc407b
commit e14b3cfbbe

View file

@ -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()