Populate tt.extern_elementwise conversion with math functions

microsoft / triton-shared

Shared Middle-Layer for Triton Compilation

MIT License

134 stars 26 forks source link

Populate tt.extern_elementwise conversion with math functions #89

Closed aobolensk closed 5 months ago

aobolensk commented 5 months ago

Add fabs, sin, cos, tan, log, log10, log1p, exp, exp2, erf, sqrt, rsqrt, ceil, floor, floor, trunc math dialect functions conversion

nhat-nguyen commented 5 months ago

@aobolensk Thank you! We're in the middle of a rewrite to split the pass into several modular passes. This code got moved to include/triton-shared/Conversion/TritonArithToLinalg/ConversionPatterns.hpp. Would appreciate if you could update your branch to latest and move the patterns over to that file.

aobolensk commented 5 months ago

@aobolensk Thank you! We're in the middle of a rewrite to split the pass into several modular passes. This code got moved to include/triton-shared/Conversion/TritonArithToLinalg/ConversionPatterns.hpp. Would appreciate if you could update your branch to latest and move the patterns over to that file.

Done