11 Commits

Author	SHA1	Message	Date
Lena Martens	7b5b9cefbd	Add scatter OOB error.	2022-01-07 17:22:34 +00:00
Lena Martens	bbd127f8fa	Add division by zero check.	2022-01-04 19:17:08 +00:00
Matthew Johnson	9e28bd5f4e	small changes to checkify ... Co-authored-by: Lena Martens <lenamartens@google.com>	2021-12-22 12:24:00 -08:00
Lena Martens	03e0deac04	Add NaN checkify rule to all lax primitives.	2021-12-20 17:51:24 +00:00
Lena Martens	98a5461132	Make sure while_loop cond generates an error even if it returns False.	2021-12-16 21:50:45 +00:00
Lena Martens	0dc5a33a88	Add checkify rule for while_loop.	2021-12-08 19:34:20 +00:00
Lena Martens	bbf1a9ba97	Add checkify rule for scan.	2021-12-07 12:36:44 +00:00
Matthew Johnson	c1f71d17c0	generalize assert primitive, allow recharging	2021-12-02 14:35:23 -08:00
Matthew Johnson	768b076420	add an assert primitive ... The assert primitive has an effectful API and so it can't be staged out; it's only a trace-time primitive. It can be discharged to the functional form. We might want to have separate transforms for discharging errors and for adding error checks. But right now they're just bundled together in the checkify transform.	2021-12-02 11:33:56 -08:00
jax authors	7869a6cb75	Merge pull request #8753 from mattjj:checkify ... PiperOrigin-RevId: 413513067	2021-12-01 14:34:17 -08:00
Matthew Johnson	659f8b794f	add skeleton checkify transformation	2021-12-01 10:44:58 -08:00