colin
/
mlir-hlo
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity
1,075 Commits 1 Branch 0 Tags 2.7 MiB
Commit Graph

3 Commits

Author SHA1 Message Date
Rahul Joshi a6011d0279 [HLO] Add AllReduceScatter to MHLO and LMHLO dialects. 
PiperOrigin-RevId: 379296198
 2021-06-14 09:37:07 -07:00
Rahul Joshi a361253e4f [HLO] Add custom print/parse for window attributes of convolutions (in LMHLO) 
PiperOrigin-RevId: 373807616
 2021-05-14 09:47:25 -07:00
Rahul Joshi 9902e6ee32 [HLO] Add LMHLO CollectivePermute verification. 
- Extract verification of source target pairs attached to collective permute into a common
  helper function and use that to verify both MHLO and LMHLO variants.
- Change MlirGpuTestBase::ParseMlirModule to allow returning back a failure, and use
  that to update the mlir_gpu_compile_test to check the new behavior.

PiperOrigin-RevId: 362156962
 2021-03-10 15:37:12 -08:00