8000 Internal change · metacortex/tensorflow@5f61825 · GitHub
[go: up one dir, main page]

Skip to content

Commit 5f61825

Browse files
pifon2atensorflower-gardener
authored andcommitted
Internal change
PiperOrigin-RevId: 334110880 Change-Id: I65c5979ccea61fd93f8f2bcb98f0f8b884d2fbc3
1 parent 34682b7 commit 5f61825

File tree

1 file changed

+9
-3
lines changed

1 file changed

+9
-3
lines changed

third_party/mlir/BUILD

Lines changed: 9 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -3104,10 +3104,10 @@ cc_binary(
31043104
],
31053105
)
31063106

3107-
cc_binary(
3108-
name = "tools/libcuda-runtime-wrappers.so",
3107+
cc_library(
3108+
name = "tools/libcuda-runtime-wrappers",
31093109
srcs = ["tools/mlir-cuda-runner/cuda-runtime-wrappers.cpp"],
3110-
linkshared = True,
3110+
compatible_with = ["//buildenv/target:prod"],
31113111
deps = [
31123112
":mlir_c_runner_utils",
31133113
"//third_party/gpus/cuda:cuda_headers",
@@ -3117,6 +3117,12 @@ cc_binary(
31173117
],
31183118
)
31193119

3120+
cc_binary(
3121+
name = "tools/libcuda-runtime-wrappers.so",
3122+
linkshared = True,
3123+
deps = [":tools/libcuda-runtime-wrappers"],
3124+
)
3125+
31203126
cc_library(
31213127
name = "VulkanRuntime",
31223128
srcs = [

0 commit comments

Comments
 (0)
0