Commit 0eba6b6b authored by A. Unique TensorFlower's avatar A. Unique TensorFlower Committed by TensorFlower Gardener
Browse files

[XLA:GPU] Support BF16 data type. 

Add an HLO pass to the GPU backend to implement BF16 operations with F32
operations.

Define macro XLA_BACKEND_SUPPORTS_BFLOAT16=1 when building tests for the GPU
backend to enable BF16 tests for GPU. Enable bfloat16_test and other BF16 tests
for GPU.

Add hlo_element_type_converter_test.

Add convolution tests and matrix multiplication tests for BF16.

PiperOrigin-RevId: 182977358
parent 9f389fc4
Show whitespace changes
Inline Side-by-side
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Please to comment