avoid CUDA init when getting GPU memory - SemanticDiff

Commit

3 years ago

avoid CUDA init when getting GPU memory

References

#1992 - [inference] test suite for ds-kernels (bert, roberta, gpt2, gpt-neo, gpt-j)

Author

mrwyattii

mrwyattii

Parents

Loading