PyTorch example

[msaroufim]

Description

This PR provides an end to end example of integrating PyTorch with cuda-python. In particular this PR

1. Ensures cuda-python reuses the same CUDA stream as PyTorch similarly to what Triton does. This logic is likely useful enough to encapsulate into its own helper on the core Stream object in cuda-python
2. Shows how to pass in pointers to torch tensors to a kernel

Limitation

1. scalars needs to be encapsulated in a torch.Tensor otherwise I was running into correctness issues

Example of running this locally

(cuda-p) [marksaroufim@devvm3083]~/cuda-python/cuda_core/examples% python pytorch_example.py 
PyTorch stream: <torch.cuda.Stream device=cuda:0 cuda_stream=0x0>
Using PyTorch stream ID: 0
Successfully created CUDA Python stream from PyTorch stream
Single precision test passed!
Double precision test passed!
All tests passed successfully!

Checklist

• New or existing tests cover these changes.
• The documentation is up to date with these changes.

[copy-pr-bot]

This pull request requires additional validation before any workflows can run on NVIDIA's runners.

Pull request vetters can view their responsibilities here.

Contributors can view more details about this message here.

[leofang]

Many thanks to your contributions, @msaroufim! 🙏 Please give us a bit of time to unblock you -- This repository, including cuda.core, originally was not allowed to accept external contributions. However, we've received legal approval to make changes for cuda.core, and #583 is the last step to open up the gate. Once we finalize the license change we can review/merge your PRs!

[msaroufim]

No worries @leofang I understand, great repo! So looking forward to contributing more soon!