| 文件 | 最后提交记录 | 最后更新时间 |
|---|---|---|
| 10 个月前 | ||
[mlir][x86vector] Lower vector.contract to FMA or packed type dot-product (#168074) A transform pass to lower vector.contract to (a) vector.fma for F32, (b) x86vector.avx512.dot for BF16, (c) x86vector.avx.dot.i8 for Int8 packed types. The lowering works on condition with m, batch, k dims to be one and vnni dim should be 2 for bf16; 4 for int8. **The lowering pattern**: batch_reduce.matmul (input) -> register-tiling(M, N) -> Vectorization (to vector.contract) -> unroll vector.contract (unit dims) -> hoisting transformation (move C loads/store outside batch/k loop) -> apply licm, canonicalization, and bufferize. | 8 个月前 | |
| 8 个月前 | ||
[mlir][x86vector] Lower vector.contract to FMA or packed type dot-product (#168074) A transform pass to lower vector.contract to (a) vector.fma for F32, (b) x86vector.avx512.dot for BF16, (c) x86vector.avx.dot.i8 for Int8 packed types. The lowering works on condition with m, batch, k dims to be one and vnni dim should be 2 for bf16; 4 for int8. **The lowering pattern**: batch_reduce.matmul (input) -> register-tiling(M, N) -> Vectorization (to vector.contract) -> unroll vector.contract (unit dims) -> hoisting transformation (move C loads/store outside batch/k loop) -> apply licm, canonicalization, and bufferize. | 8 个月前 |
| 文件 | 最后提交记录 | 最后更新时间 |
|---|---|---|
| 10 个月前 | ||
| 8 个月前 | ||
| 8 个月前 | ||
| 8 个月前 |