Preliminary HIP bfloat16 GPU support.

This commit is contained in:
Rohit Santhanam
2021-10-27 18:36:45 +00:00
committed by Antonio Sánchez
parent 40bbe8a4d0
commit 48e40b22bf
4 changed files with 621 additions and 17 deletions

View File

@@ -385,6 +385,7 @@ if (EIGEN_TEST_HIP)
ei_add_test(cxx11_tensor_gpu)
ei_add_test(cxx11_tensor_contract_gpu)
ei_add_test(cxx11_tensor_of_float16_gpu)
ei_add_test(cxx11_tensor_of_bfloat16_gpu)
ei_add_test(cxx11_tensor_random_gpu)
unset(EIGEN_ADD_TEST_FILENAME_EXTENSION)