Update pytorch_example.py

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

Author: msaroufim

cuda_core/examples/pytorch_example.py

@@ -73,9 +73,6 @@ def __cuda_stream__(self):
 # launch kernel on our 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("Single precision test passed!")