From 1278503a1c859d557174a4ef2ae7a85295f39f69 Mon Sep 17 00:00:00 2001
From: Willow Ahrens <willow.marie.ahrens@gmail.com>
Date: Thu, 5 Oct 2023 12:33:39 -0400
Subject: [PATCH] make benchmarking functions public

---
 include/taco/tensor.h | 9 +++++----
 1 file changed, 5 insertions(+), 4 deletions(-)

diff --git a/include/taco/tensor.h b/include/taco/tensor.h
index c462cbd32..dbc1cf339 100644
--- a/include/taco/tensor.h
+++ b/include/taco/tensor.h
@@ -445,6 +445,11 @@ class TensorBase {
   /// True if the Tensor needs to be computed.
   bool needsCompute();
 
+  void setNeedsPack(bool needsPack);
+  void setNeedsCompile(bool needsCompile);
+  void setNeedsAssemble(bool needsAssemble);
+  void setNeedsCompute(bool needsCompute);
+
   /// Set to true to perform the assemble and compute stages simultaneously.
   void setAssembleWhileCompute(bool assembleWhileCompute);
 
@@ -510,10 +515,6 @@ class TensorBase {
   bool neverPacked();
 
   void unsetNeverPacked();
-  void setNeedsPack(bool needsPack);
-  void setNeedsCompile(bool needsCompile);
-  void setNeedsAssemble(bool needsAssemble);
-  void setNeedsCompute(bool needsCompute);
 
   void addDependentTensor(TensorBase& tensor);
   void removeDependentTensor(TensorBase& tensor);