Update cuda_core/examples/pytorch_example.py

Co-authored-by: Leo Fang <leo80042@gmail.com>

Author: msaroufim

cuda_core/examples/pytorch_example.py

@@ -100,9 +100,6 @@ def __cuda_stream__(self):
 # launch kernel on PyTorch's stream
 launch(s, config, ker, *ker_args)
 
-# Wait for our CUDA kernel to complete
-s.sync()
-
 # check result
 assert torch.allclose(out, a.item() * x + y)
 print("Double precision test passed!")