[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.
2018-12-13 15:09:54 -08:00 · 2018-12-13 15:09:54 -08:00 · 1674bd08ce
commit 1674bd08ce
parent 1182414979
1 changed files with 1 additions and 0 deletions
--- 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<int, int> compute_capability) {
      {{6, 2}, 62},
      {{7, 0}, 70},
      {{7, 2}, 72},
+      {{7, 5}, 75},
  });
  int sm_version = 30;
  auto it = m->find(compute_capability);