Skip to content

Commit 8e4d126

Browse files
committed
Switch the arguments order
1 parent 79dd27c commit 8e4d126

File tree

3 files changed

+3
-3
lines changed

3 files changed

+3
-3
lines changed

include/taco/index_notation/index_notation.h

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -786,7 +786,7 @@ class IndexStmt : public util::IntrusivePtr<const IndexStmtNode> {
786786
///
787787
/// Precondition:
788788
/// Workspace can be accessed by the IndexVars in the accels.
789-
IndexStmt wsaccel(TensorVar& ws, const std::vector<IndexVar>& accels, bool Acc = true);
789+
IndexStmt wsaccel(TensorVar& ws, bool Acc = true,const std::vector<IndexVar>& accels={});
790790

791791
/// Casts index statement to specified subtype.
792792
template <typename SubType>

src/index_notation/index_notation.cpp

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -2048,7 +2048,7 @@ IndexStmt IndexStmt::assemble(TensorVar result, AssembleStrategy strategy,
20482048
return transformed;
20492049
}
20502050

2051-
IndexStmt IndexStmt::wsaccel(TensorVar& ws, const std::vector<IndexVar>& accels, bool Acc) {
2051+
IndexStmt IndexStmt::wsaccel(TensorVar& ws, bool Acc, const std::vector<IndexVar>& accels) {
20522052
if (accels.size() == 0) {
20532053
ws.setAccels(accels, Acc);
20542054
return *this;

test/tests-workspaces.cpp

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -546,7 +546,7 @@ TEST(workspaces, tile_dotProduct_2) {
546546

547547
stmt = stmt.concretize();
548548

549-
stmt = stmt.wsaccel(precomputed, {}, false);
549+
stmt = stmt.wsaccel(precomputed, false);
550550
A.compile(stmt);
551551
A.assemble();
552552
A.compute();

0 commit comments

Comments
 (0)