From 1674bd08ce0c49a75a0f01c18ef940fa012124a8 Mon Sep 17 00:00:00 2001 From: Ben Barsdell Date: Thu, 13 Dec 2018 15:09:54 -0800 Subject: [PATCH] [XLA:GPU] Add sm_75 to LLVM supported procs list - This is required to prevent fallback to the default sm version when running on GPUs with the Turing architecture. --- .../xla/service/gpu/llvm_gpu_backend/nvptx_backend_lib.cc | 1 + 1 file changed, 1 insertion(+) diff --git a/tensorflow/compiler/xla/service/gpu/llvm_gpu_backend/nvptx_backend_lib.cc b/tensorflow/compiler/xla/service/gpu/llvm_gpu_backend/nvptx_backend_lib.cc index bd53b90b42d..eddaa877f28 100644 --- a/tensorflow/compiler/xla/service/gpu/llvm_gpu_backend/nvptx_backend_lib.cc +++ b/tensorflow/compiler/xla/service/gpu/llvm_gpu_backend/nvptx_backend_lib.cc @@ -125,6 +125,7 @@ static string GetSmName(std::pair compute_capability) { {{6, 2}, 62}, {{7, 0}, 70}, {{7, 2}, 72}, + {{7, 5}, 75}, }); int sm_version = 30; auto it = m->find(compute_capability);