Enable cross entropy loss for xla autocast with FP32 precision (#7992) #10096
Triggered via pull request
September 28, 2024 23:21
avizon-aws
Status
Success
Total duration
1h 22m 21s
Artifacts
5
build_and_test.yml
on: pull_request
get-torch-commit
1s
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 |
|