Skip to content

Commit e999e44

Browse files
authored
Update BUILD with if_cuda_is_configured
1 parent af8e82d commit e999e44

File tree

1 file changed

+4
-3
lines changed
  • tensorflow_quantum/core/ops

1 file changed

+4
-3
lines changed

tensorflow_quantum/core/ops/BUILD

Lines changed: 4 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -43,15 +43,16 @@ py_library(
4343
":tfq_adj_grad_op_py",
4444
":tfq_ps_util_ops_py",
4545
":tfq_simulate_ops_py",
46-
":tfq_simulate_ops_cuda_py",
47-
":tfq_simulate_ops_cuquantum_py",
4846
":tfq_unitary_op_py",
4947
":tfq_utility_ops_py",
5048
# test addons
5149
"//tensorflow_quantum/core/ops/math_ops:inner_product_op_py",
5250
"//tensorflow_quantum/core/ops/math_ops:fidelity_op_py",
5351
"//tensorflow_quantum/core/ops/noise:noisy_expectation_op_py",
54-
],
52+
] + if_cuda_is_configured([
53+
":tfq_simulate_ops_cuda_py",
54+
":tfq_simulate_ops_cuquantum_py",
55+
]),
5556
)
5657

5758
cc_binary(

0 commit comments

Comments
 (0)