[Cherry-pick] Enable cross entropy loss for xla autocast with FP32 pr… #10118
build_and_test.yml
on: push
get-torch-commit
1s
Build XLA CUDA plugin
/
build
5m 25s
TPU tests
/
tpu-test
34m 7s
Build docs
/
build-docs
1m 36s
Matrix: GPU tests / test
Matrix: CPU tests / test
Matrix: GPU tests requiring torch CUDA / test
Artifacts
Produced during runtime
Name | Size | |
---|---|---|
cpp-test-bin
Expired
|
699 MB |
|
cuda-plugin
Expired
|
130 MB |
|
torch-with-cuda
Expired
|
341 MB |
|
torch-xla-wheels
Expired
|
217 MB |
|