File tree Expand file tree Collapse file tree 1 file changed +6
-3
lines changed Expand file tree Collapse file tree 1 file changed +6
-3
lines changed Original file line number Diff line number Diff line change
1
+ load ("@local_config_cuda//cuda:build_defs.bzl" , "if_cuda_is_configured" )
2
+
1
3
licenses (["notice" ])
2
4
3
5
sh_binary (
@@ -39,8 +41,6 @@ sh_binary(
39
41
"//tensorflow_quantum/core/ops:tfq_unitary_op_py" ,
40
42
"//tensorflow_quantum/core/ops:tfq_utility_ops_py" ,
41
43
"//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" ,
44
44
"//tensorflow_quantum/core/ops/math_ops:fidelity_op_py" ,
45
45
"//tensorflow_quantum/core/ops/math_ops:inner_product_op_py" ,
46
46
"//tensorflow_quantum/core/ops/math_ops:simulate_mps_py" ,
@@ -68,5 +68,8 @@ sh_binary(
68
68
"//tensorflow_quantum/python:util" ,
69
69
"//tensorflow_quantum/python/optimizers:rotosolve_minimizer" ,
70
70
"//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
+ ]),
72
75
)
You can’t perform that action at this time.
0 commit comments