Skip to content

Commit bb2c591

Browse files
authored
kFloat->kComplexFloat in benchmarks to allow running in debug build (#156)
1 parent 001612e commit bb2c591

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

benchmarks/benchmarks.cpp

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff 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);

0 commit comments

Comments
 (0)