Skip to content

Commit f2cd255

Browse files
authored
Update release/BUILD with if_cuda_is_configured
1 parent e999e44 commit f2cd255

File tree

1 file changed

+6
-3
lines changed

1 file changed

+6
-3
lines changed

release/BUILD

Lines changed: 6 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -1,3 +1,5 @@
1+
load("@local_config_cuda//cuda:build_defs.bzl", "if_cuda_is_configured")
2+
13
licenses(["notice"])
24

35
sh_binary(
@@ -39,8 +41,6 @@ sh_binary(
3941
"//tensorflow_quantum/core/ops:tfq_unitary_op_py",
4042
"//tensorflow_quantum/core/ops:tfq_utility_ops_py",
4143
"//tensorflow_quantum/core/ops:tfq_simulate_ops_py",
42-
"//tensorflow_quantum/core/ops:tfq_simulate_ops_cuda_py",
43-
"//tensorflow_quantum/core/ops:tfq_simulate_ops_cuquantum_py",
4444
"//tensorflow_quantum/core/ops/math_ops:fidelity_op_py",
4545
"//tensorflow_quantum/core/ops/math_ops:inner_product_op_py",
4646
"//tensorflow_quantum/core/ops/math_ops:simulate_mps_py",
@@ -68,5 +68,8 @@ sh_binary(
6868
"//tensorflow_quantum/python:util",
6969
"//tensorflow_quantum/python/optimizers:rotosolve_minimizer",
7070
"//tensorflow_quantum/python/optimizers:spsa_minimizer",
71-
],
71+
] + if_cuda_is_configured([
72+
"//tensorflow_quantum/core/ops:tfq_simulate_ops_cuda_py",
73+
"//tensorflow_quantum/core/ops:tfq_simulate_ops_cuquantum_py",
74+
]),
7275
)

0 commit comments

Comments
 (0)