pykokkos icon indicating copy to clipboard operation
pykokkos copied to clipboard

Add support for passing pytorch tensors directly to kernel calls

Open NaderAlAwar opened this issue 7 months ago • 2 comments

This checks if torch is installed and uses torch.is_tensor() to check if the passed argument is a pytorch tensor.

NaderAlAwar avatar Jul 07 '24 10:07 NaderAlAwar