pymlg icon indicating copy to clipboard operation
pymlg copied to clipboard

added cuda support for all pytorch mlg operations

Open angadbajwa opened this issue 7 months ago • 1 comments

tests passing on a GeForce GTX 1650 Mobile / Max-Q w/ CUDA 12.8

  • [ ] Still need to add complex tensor/ndarray support for complex step jacobian. not currently sure how to test that?

angadbajwa avatar Sep 02 '25 14:09 angadbajwa

@angadbajwa thanks for doing this man! the automated testing environment doesn't seem to have a GPU, which will cause the unit tests to fail. I suppose you could just use torch.cuda.is_available in the unit tests themselves for now, so that it doesn't run in the github environment. not ideal but good enough

CharlesCossette avatar Nov 06 '25 19:11 CharlesCossette