Enable cross entropy loss for xla autocast with FP32 precision (#7992… #10103
lsy323build_and_test.yml
on: push
get-torch-commit
7s
Build XLA CUDA plugin
/
build
5m 25s
Matrix: GPU tests / test
Matrix: CPU tests / test
Matrix: GPU tests requiring torch CUDA / test
Artifacts
Produced during runtime
| Name | Size | |
|---|---|---|
|
cpp-test-bin
|
699 MB |
|
|
cuda-plugin
|
130 MB |
|
|
github-pages
|
4.09 MB |
|
|
torch-with-cuda
|
342 MB |
|
|
torch-xla-wheels
|
217 MB |
|