Skip to content

Commit b0788bb

Browse files
committed
Fix names
1 parent f8ce3de commit b0788bb

File tree

4 files changed

+11
-11
lines changed

4 files changed

+11
-11
lines changed

include/taco/index_notation/index_notation.h

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -1188,13 +1188,13 @@ class TensorVar : public util::Comparable<TensorVar> {
11881188
const Literal& getFill() const;
11891189

11901190
/// Gets the acceleration dimensions
1191-
const std::vector<IndexVar>& getaccelIndexVars() const;
1191+
const std::vector<IndexVar>& getAccelIndexVars() const;
11921192

11931193
/// Gets the acceleration flag
1194-
bool getshouldAccel() const;
1194+
bool getShouldAccel() const;
11951195

11961196
/// Set the acceleration dimensions
1197-
void setaccelIndexVars(const std::vector<IndexVar>& accelIndexVars, bool shouldAccel);
1197+
void setAccelIndexVars(const std::vector<IndexVar>& accelIndexVars, bool shouldAccel);
11981198

11991199
/// Set the fill value of the tensor variable
12001200
void setFill(const Literal& fill);

src/index_notation/index_notation.cpp

Lines changed: 5 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -2050,7 +2050,7 @@ IndexStmt IndexStmt::assemble(TensorVar result, AssembleStrategy strategy,
20502050

20512051
IndexStmt IndexStmt::wsaccel(TensorVar& ws, bool shouldAccel, const std::vector<IndexVar>& accelIndexVars) {
20522052
if (accelIndexVars.size() == 0) {
2053-
ws.setaccelIndexVars(accelIndexVars, shouldAccel);
2053+
ws.setAccelIndexVars(accelIndexVars, shouldAccel);
20542054
return *this;
20552055
}
20562056
set<IndexVar> TempVars;
@@ -2070,7 +2070,7 @@ IndexStmt IndexStmt::wsaccel(TensorVar& ws, bool shouldAccel, const std::vector<
20702070
taco_uerror << "No matching indexVars in the Accel";
20712071
}
20722072
}
2073-
ws.setaccelIndexVars(accelIndexVars, shouldAccel);
2073+
ws.setAccelIndexVars(accelIndexVars, shouldAccel);
20742074
return *this;
20752075
}
20762076

@@ -2625,15 +2625,15 @@ const Literal& TensorVar::getFill() const {
26252625
return content->fill;
26262626
}
26272627

2628-
const std::vector<IndexVar>& TensorVar::getaccelIndexVars() const {
2628+
const std::vector<IndexVar>& TensorVar::getAccelIndexVars() const {
26292629
return content->accelIndexVars;
26302630
}
26312631

2632-
bool TensorVar::getshouldAccel() const {
2632+
bool TensorVar::getShouldAccel() const {
26332633
return content->shouldAccel;
26342634
}
26352635

2636-
void TensorVar::setaccelIndexVars(const std::vector<IndexVar>& accelIndexVars, bool shouldAccel) {
2636+
void TensorVar::setAccelIndexVars(const std::vector<IndexVar>& accelIndexVars, bool shouldAccel) {
26372637
content->shouldAccel = shouldAccel;
26382638
content->accelIndexVars = accelIndexVars;
26392639
}

src/lower/lowerer_impl_imperative.cpp

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -2276,7 +2276,7 @@ std::pair<bool,bool> LowererImplImperative::canAccelerateDenseTemp(Where where)
22762276
TensorVar temporary = where.getTemporary();
22772277

22782278
// (0) Acceleration flag is true
2279-
if (!temporary.getshouldAccel()) {
2279+
if (!temporary.getShouldAccel()) {
22802280
return std::make_pair(false, false);
22812281
}
22822282

test/tests-scheduling-eval.cpp

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -658,8 +658,8 @@ INSTANTIATE_TEST_CASE_P(spgemm, spgemm,
658658
Values(std::make_tuple(CSR, CSR, true),
659659
std::make_tuple(DCSR, CSR, true),
660660
std::make_tuple(DCSR, DCSR, true),
661-
std::make_tuple(CSR, CSC, false),
662-
std::make_tuple(DCSR, DCSC, false)));
661+
std::make_tuple(CSR, CSC, true),
662+
std::make_tuple(DCSR, DCSC, true)));
663663

664664
TEST(scheduling_eval, spmataddCPU) {
665665
if (should_use_CUDA_codegen()) {

0 commit comments

Comments
 (0)