hikettei / Caten

[wip] Deep Learning Compiler based on Polyhedral Compiler and Light-weight IRs based on Optimizing Pattern Matcher
https://hikettei.github.io/Caten/
Other
20 stars 4 forks source link

BugFix: Tensor Shaped Index-Components #240

Closed hikettei closed 4 days ago

hikettei commented 4 days ago
hikettei commented 4 days ago

but need to remove a ton of unused constants?

void fused_maxreduce_maxreduce_argmaxnode263849(int64_t* val_1536, const int64_t val_1534, const int64_t val_1494, const int64_t val_1526, const int64_t val_1498, const int64_t val_1521, const int64_t val_1508, const int64_t val_1505, const int64_t val_1472, const float* restrict val_1449) {
  float val_1495 = 32.0;
  float val_1485 = 0.0;
  float val_1488 = 0.0;
  for (int _gid2=0; (_gid2<32); _gid2+=1) {
    val_1488 = max(val_1488, val_1449[(((128*s)*_gid2)+(128*val_1472))]);
  }
  for (int _gid2=0; (_gid2<32); _gid2+=1) {
    val_1485 = max(val_1485, (!((val_1449[(((128*s)*_gid2)+(128*val_1472))]!=val_1488)) ? (31.0+(0*-1.0)) : 0.0));
  }
  val_1536[0] = (int64_t)((val_1495+-(val_1485))+-1.0);
}