From 149a0a1d5a1b920568b8148a31c57903b116e80a Mon Sep 17 00:00:00 2001
From: Adrian Kuegel <akuegel@google.com>
Date: Mon, 22 Jun 2020 01:38:54 -0700
Subject: [PATCH] Add support for windows.
The tools from the cuda toolkit have the .exe extension on Windows.
PiperOrigin-RevId: 317610348
Change-Id: I6f4bad4c651c8ffb347f06d1aef8ffe00de90620
---
tensorflow/core/kernels/cubin_headers/build_defs.bzl | 2 +-
1 file changed, 1 insertion(+), 1 deletion(-)
diff --git a/tensorflow/core/kernels/cubin_headers/build_defs.bzl b/tensorflow/core/kernels/cubin_headers/build_defs.bzl
index 5880cbe8add..c3e44b7a974 100644
--- a/tensorflow/core/kernels/cubin_headers/build_defs.bzl
+++ b/tensorflow/core/kernels/cubin_headers/build_defs.bzl
@@ -5,7 +5,7 @@ load("@local_config_cuda//cuda:build_defs.bzl", "cuda_gpu_architectures", "if_cu
def _lookup_file(filegroup, path):
"""Extracts file at (relative) path in filegroup."""
for file in filegroup.files.to_list():
- if file.path.endswith(path):
+ if file.path.endswith(path) or file.path.endswith(path + ".exe"):
return file
return None