squall/cutlass
1
0
Fork 0
Code Issues Pull Requests Actions 1 Packages Projects Releases Wiki Activity

change unused class member to local var (#646)

Browse Source
This commit is contained in:
Wenzhuo Liu 2022-09-29 11:52:35 +08:00 committed by GitHub
parent 48a9ea223a
commit cd37e82492
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 10 additions and 19 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

27
include/cutlass/layout/permute.h
View File

@ -119,9 +119,6 @@ private:
Index stride_permute_; Index stride_permute_;
Index col_permute_;
Index row_permute_;
public: public:
// //
// Methods // Methods
@ -159,10 +156,10 @@ public:
int i = row_init / D1; int i = row_init / D1;
// After the Permute Op // After the Permute Op
col_permute_ = l + j * D3; Index col_permute = l + j * D3;
row_permute_ = k + i * D2; Index row_permute = k + i * D2;
return LongIndex(row_permute_) * LongIndex(stride_permute_) + LongIndex(col_permute_); return LongIndex(row_permute) * LongIndex(stride_permute_) + LongIndex(col_permute);
} }
/// Return D1 /// Return D1
@ -198,9 +195,6 @@ private:
Index stride_permute_; Index stride_permute_;
Index col_permute_;
Index row_permute_;
public: public:
// //
// Methods // Methods
@ -240,10 +234,10 @@ public:
int i = BMM_batch_idx / D1; int i = BMM_batch_idx / D1;
// After the Permute Op // After the Permute Op
col_permute_ = l + j * D3; Index col_permute = l + j * D3;
row_permute_ = k + i * D2; Index row_permute = k + i * D2;
return LongIndex(row_permute_) * LongIndex(stride_permute_) + LongIndex(col_permute_); return LongIndex(row_permute) * LongIndex(stride_permute_) + LongIndex(col_permute);
} }
/// Return D1 /// Return D1
@ -273,9 +267,6 @@ private:
Index stride_permute_; Index stride_permute_;
Index col_permute_;
Index row_permute_;
public: public:
// //
// Methods // Methods
@ -313,10 +304,10 @@ public:
int i = row_init / T1; int i = row_init / T1;
// After the Permute Op // After the Permute Op
col_permute_ = m + j * T4 + l * T1 * T4; Index col_permute = m + j * T4 + l * T1 * T4;
row_permute_ = i + k * T0; Index row_permute = i + k * T0;
return LongIndex(row_permute_) * LongIndex(stride_permute_) + LongIndex(col_permute_); return LongIndex(row_permute) * LongIndex(stride_permute_) + LongIndex(col_permute);
} }
}; };