From a022b9b835a9e203f25ca6b7a9e80be45eef4724 Mon Sep 17 00:00:00 2001 From: "Wen-Heng (Jack) Chung" Date: Tue, 2 Jul 2019 09:29:24 -0500 Subject: [PATCH] Add gfx908 into supported AMD GCN ISA version list --- tensorflow/core/common_runtime/gpu/gpu_device.cc | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) 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;