From 9135a963c01d97ba34b1a7d2f106e78a13fd6651 Mon Sep 17 00:00:00 2001
From: "Gao, Xiang" <qasdfgtyuiop@gmail.com>
Date: Sat, 16 Jul 2022 06:23:53 -0700
Subject: [PATCH] Fix transpose benchmark dtype (#1839)

---
 benchmarks/cpp/nvfuser/transpose.cpp | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/benchmarks/cpp/nvfuser/transpose.cpp b/benchmarks/cpp/nvfuser/transpose.cpp
index b6cfdbc70a20f..ec6738303a3fc 100644
--- 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);