search

cresset-template / cresset

Template repository to build PyTorch projects from source on any version of PyTorch/CUDA/cuDNN.
MIT License
714 stars 41 forks source link

Update tests #160

Closed veritas9872 closed 7 months ago

veritas9872 commented 10 months ago

Update tests for PyTorch 2.x and CUDA 12.x. Also get started with Transformer Engine, etc.

veritas9872 commented 9 months ago

Add DeepSpeed and Transformers. Not so sure about Transformer Engine.

veritas9872 commented 7 months ago

Decided against updating as the tests still work fine.