made the function to place data on the GPU public

fidelity / stoke

A lightweight wrapper for PyTorch that provides a simple declarative API for context switching between devices, distributed modes, mixed-precision, and PyTorch extensions.

https://fidelity.github.io/stoke/

Apache License 2.0

66 stars 3 forks source link

made the function to place data on the GPU public #57

Closed ncilfone closed 2 years ago

ncilfone commented 2 years ago

What does this PR do?

made the function to place data on the GPU public

Checklist

[x] Did you adhere to PEP-8 standards?
[x] Did you run black and isort prior to submitting your PR?
[x] Does your PR pass all existing unit tests?
[x] Did you add associated unit tests for any additional functionality?
[x] Did you provide documentation (Numpy Docstring format) whenever possible, even for simple functions or classes?