Skip to content
This repository has been archived by the owner on Sep 9, 2024. It is now read-only.

Commit

Permalink
fix CI.
Browse files Browse the repository at this point in the history
  • Loading branch information
KuangjuX committed Feb 14, 2024
1 parent 205ffa5 commit f6298a5
Showing 1 changed file with 4 additions and 4 deletions.
8 changes: 4 additions & 4 deletions src/generator.cc
Original file line number Diff line number Diff line change
Expand Up @@ -92,7 +92,7 @@ namespace tiledkernel {
ASSERT(access_map_rA->getLoops() == access_map_acc->getLoops(),
"Access map should have the same number of loops.");

for (auto loop = 0; loop < access_map_rA->getLoops(); loop++) {
for (size_t loop = 0; loop < access_map_rA->getLoops(); loop++) {
if (access_map_rA->hasPinIterVar(loop)) {
ASSERT(access_map_rB->hasPinIterVar(loop),
"Access map should have the same pin iter var.");
Expand Down Expand Up @@ -131,7 +131,7 @@ namespace tiledkernel {
auto kernel_body = fmt::format("gemm({}, {}, {});\n", rA_access,
rB_access, acc_access);

for (auto loop = 0; loop < access_map_rA->getLoops(); loop++) {
for (size_t loop = 0; loop < access_map_rA->getLoops(); loop++) {
kernel_body = generate_loop(
access_map_rA->pin_iter_vars[loop].value(), kernel_body);
}
Expand Down Expand Up @@ -162,10 +162,10 @@ namespace tiledkernel {
std::string TiledGenerator::generate_access_map(
AccessMap::Pointer access_map) {
std::string kernel;
for (auto access_loop = 0; access_loop < access_map->getAccessDims();
for (size_t access_loop = 0; access_loop < access_map->getAccessDims();
access_loop++) {
auto access_dim = access_map->access_pattern[access_loop];
for (auto loop = 0; loop < access_map->getLoops(); loop++) {
for (size_t loop = 0; loop < access_map->getLoops(); loop++) {
if (access_dim[loop] == 0) {
return "";
} else if (access_dim[loop] == 1) {
Expand Down

0 comments on commit f6298a5

Please sign in to comment.