File tree Expand file tree Collapse file tree 1 file changed +3
-3
lines changed
Expand file tree Collapse file tree 1 file changed +3
-3
lines changed Original file line number Diff line number Diff line change @@ -56,7 +56,7 @@ static void BM_vacuumOscillations(benchmark::State &state)
5656
5757 // make some random test energies
5858 Tensor energies =
59- Tensor::scale (Tensor::rand ({state.range (0 ), 1 }).dType (dtypes::kFloat ).requiresGrad (false ), 10000.0 ).hasBatchDim (true ) +
59+ Tensor::scale (Tensor::rand ({state.range (0 ), 1 }).dType (dtypes::kComplexFloat ).requiresGrad (false ), 10000.0 ).hasBatchDim (true ) +
6060 Tensor ({100.0 });
6161
6262 energies = energies.hasBatchDim (true );
@@ -92,7 +92,7 @@ static void BM_constMatterOscillations(benchmark::State &state)
9292
9393 // make some random test energies
9494 Tensor energies =
95- Tensor::scale (Tensor::rand ({state.range (0 ), 1 }).dType (dtypes::kFloat ).requiresGrad (false ), 10000.0 ) +
95+ Tensor::scale (Tensor::rand ({state.range (0 ), 1 }).dType (dtypes::kComplexFloat ).requiresGrad (false ), 10000.0 ) +
9696 Tensor ({100.0 });
9797
9898 energies = energies.hasBatchDim (true );
@@ -131,7 +131,7 @@ static void BM_DPpropOscillations(benchmark::State &state)
131131
132132 // make some random test energies
133133 Tensor energies =
134- Tensor::scale (Tensor::rand ({state.range (0 ), 1 }).dType (dtypes::kFloat ).requiresGrad (false ), 10000.0 ) +
134+ Tensor::scale (Tensor::rand ({state.range (0 ), 1 }).dType (dtypes::kComplexFloat ).requiresGrad (false ), 10000.0 ) +
135135 Tensor ({100.0 });
136136
137137 energies = energies.hasBatchDim (true );
You can’t perform that action at this time.
0 commit comments