Skip to content

Instantly share code, notes, and snippets.

@mlazos

mlazos/gist:4d57df4f7b43bb0531100deb7cc51e10

Created February 18, 2026 22:32
Show Gist options

  • Select an option

    Learn more about clone URLs
  • Save mlazos/4d57df4f7b43bb0531100deb7cc51e10 to your computer and use it in GitHub Desktop.

Select an option

Learn more about clone URLs
Save mlazos/4d57df4f7b43bb0531100deb7cc51e10 to your computer and use it in GitHub Desktop.
Download ZIP
Raw
gistfile1.txt
with torch.Stream(device='cuda') as s_cuda:
a = torch.randn(10, 5, device='cuda')
b = torch.randn(5, 10, device='cuda')
c = torch.mm(a, b)
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment