Enhancement: MultiExpr in the same domain - Githubissues

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

15 stars 1 forks source link

Enhancement: MultiExpr in the same domain #114

Closed hikettei closed 1 month ago

hikettei commented 1 month ago

[x] Post-MultiExpr (Completely Equivalent Domain)
[ ] Post-MultiExpr (where subdomain is the equivalent)
[ ] Index-Component-Fusion
[ ] Add the following transform pattern to EXPR:

[ ] More Restrict Schedule test in nn or ajit (e.g.: Embedding in one kernel)


val_445[(_gid0+0)] = (val_456_0^((val_445[(_gid0+0)]*32768)+(uint32_t)((float)val_445[(_gid0+0)]*7.6293945e-6)));
->
float _tmp_0 = val_445[_gid0+0];
val_445[_gid0+0] = (val_456_0^((_tmp_0*32768)+(uint32_t)((float)_tmp_0*7.6293945e-6)));
``