mlir-hlo/lib/Dialect/mhlo/IR
Smit Hinsu 9456af5880 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: 345605910
2020-12-03 22:09:56 -08:00
..
CMakeLists.txt [MLIR:HLO] Generate enum decls for HLO and LHLO GPU dialects. 2020-12-02 11:39:23 -08: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-03 22:09:56 -08:00
hlo_ops_base_enums.cc [MLIR:HLO] Generate enum decls for HLO and LHLO GPU dialects. 2020-12-02 11:39:23 -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_enums.cc [MLIR:HLO] Generate enum decls for HLO and LHLO GPU dialects. 2020-12-02 11:39:23 -08: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