Fix transpose benchmark dtype (#1839)

csarofeen · Jul 16, 2022 · 9135a96 · 9135a96
1 parent 2c9a6c0
commit 9135a96
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/benchmarks/cpp/nvfuser/transpose.cpp b/benchmarks/cpp/nvfuser/transpose.cpp
@@ -84,8 +84,8 @@ static void setupTranspose(
  return (is_transpose) ? transpose(tv, axes.first, axes.second) : tv;
  };
 
- auto input1 = makeContigTensor(num_dims);
- auto input2 = makeContigTensor(num_dims);
+ auto input1 = makeContigTensor(num_dims, dtype);
+ auto input2 = makeContigTensor(num_dims, dtype);
  fusion->addInput(input1);
  fusion->addInput(input2);