Fix test

NVIDIA · Nov 12, 2024 · 47b8ea7 · 47b8ea7
1 parent 21c6e2e
commit 47b8ea7
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/benchmarks/cpp/lstm_cell.cpp b/benchmarks/cpp/lstm_cell.cpp
@@ -27,11 +27,11 @@ static void setupFusion(Fusion* fusion) {
 
   TensorView* tvs[16];
   for (size_t i = 0; i < 16; i++) {
-    tvs[i] = makeContigTensor(2, DataType::Float);
+    tvs[i] = makeSymbolicTensor(2, DataType::Float);
     fusion->addInput(tvs[i]);
   }
 
-  const auto cx = makeContigTensor(2, DataType::Float);
+  const auto cx = makeSymbolicTensor(2, DataType::Float);
   fusion->addInput(cx);
 
   const auto in_x = add(add(add(tvs[0], tvs[1]), tvs[2]), tvs[3]);