Skip to content

Commit 0e79acb

Browse files
Merge pull request #574 from tensor-compiler/benchmark
make benchmarking functions public
2 parents a3d8792 + 9634fab commit 0e79acb

File tree

1 file changed

+5
-4
lines changed

1 file changed

+5
-4
lines changed

include/taco/tensor.h

Lines changed: 5 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -445,6 +445,11 @@ class TensorBase {
445445
/// True if the Tensor needs to be computed.
446446
bool needsCompute();
447447

448+
void setNeedsPack(bool needsPack);
449+
void setNeedsCompile(bool needsCompile);
450+
void setNeedsAssemble(bool needsAssemble);
451+
void setNeedsCompute(bool needsCompute);
452+
448453
/// Set to true to perform the assemble and compute stages simultaneously.
449454
void setAssembleWhileCompute(bool assembleWhileCompute);
450455

@@ -510,10 +515,6 @@ class TensorBase {
510515
bool neverPacked();
511516

512517
void unsetNeverPacked();
513-
void setNeedsPack(bool needsPack);
514-
void setNeedsCompile(bool needsCompile);
515-
void setNeedsAssemble(bool needsAssemble);
516-
void setNeedsCompute(bool needsCompute);
517518

518519
void addDependentTensor(TensorBase& tensor);
519520
void removeDependentTensor(TensorBase& tensor);

0 commit comments

Comments
 (0)