diff --git a/tensorflow/core/common_runtime/gpu/gpu_device.cc b/tensorflow/core/common_runtime/gpu/gpu_device.cc index 05f90e70f3d..796dbbbbda2 100644 --- a/tensorflow/core/common_runtime/gpu/gpu_device.cc +++ b/tensorflow/core/common_runtime/gpu/gpu_device.cc @@ -1542,7 +1542,7 @@ std::vector GetSupportedCudaComputeCapabilities() { #endif // GOOGLE_CUDA #if TENSORFLOW_USE_ROCM -std::vector supported_amdgpu_isa_versions = {803, 900, 906}; +std::vector supported_amdgpu_isa_versions = {803, 900, 906, 908}; std::vector GetSupportedAMDGPUISAVersions() { return supported_amdgpu_isa_versions;