mlir-hlo/lib/Dialect/mhlo/IR
Smit Hinsu 733fc6d032 Fix handling of negative seeds in random number generator op kernels for XLA
Casting negative s32 number to u64 directly will have leading 1s in the representation which is not what we want to get a single u64 out of two s32 seeds. Fixed this by first getting unsigned number of the same bit-width.

PiperOrigin-RevId: 345227848
2020-12-02 07:24:10 -08:00
..
CMakeLists.txt Make LMHLO's Dot have the same power as MHLO's DotGeneral. 2020-10-15 15:09:06 -07:00
chlo_canonicalize.td Add chlo.constant_like op which splats a constant to shape of operand 2020-08-11 14:54:48 -07:00
chlo_ops.cc Add compare_type optional attribute to CompareOp in HLO dialects 2020-10-26 12:58:29 -07:00
hlo_ops.cc Fix handling of negative seeds in random number generator op kernels for XLA 2020-12-02 07:24:10 -08:00
hlo_ops_base_structs.cc Make LMHLO's Dot have the same power as MHLO's DotGeneral. 2020-10-15 15:09:06 -07:00
hlo_patterns.td Add canonicalizer for Reshape(Broadcast(X)) pattern when it is an identity sequence 2020-11-19 02:32:45 -08:00
infer_fusibility_op_interface.cc More cleanup in mlir-hlo to prepare for the standalone build 2020-08-03 19:28:00 -07:00
init.cc Add GPU specific LMHLO level ops 2020-10-14 11:23:55 -07:00
lhlo_gpu_ops.cc Make LMHLO's Dot have the same power as MHLO's DotGeneral. 2020-10-15 15:09:06 -07:00
lhlo_gpu_ops_structs.cc Make LMHLO's Dot have the same power as MHLO's DotGeneral. 2020-10-15 15:09:06 -07:00
lhlo_ops.cc [HLO] Delete LHLO memref cast ops and migrate to STD ones. 2020-11-04 09:26:34 -08:00
lhlo_ops_structs.cc Make LMHLO's Dot have the same power as MHLO's DotGeneral. 2020-10-15 15:09:06 -07:00
mhlo_canonicalize.td Add a canonicalization pattern to remove redundant dynamic_reshapes. 2020-11-27 04:46:50 -08:00