6 Commits (464feacba9b01882bb32093034c749814f9728e2)

Author	SHA1	Message	Date
Sean Silva	703428eff4	Add support for "trailing_" and "out" variants of various ops. ... We already had the `promoteTrailingOutTensor` flag, but weren't using it. A inplaceVariantKernelName flag needed to be added. This change is a little dissatisfying, as the conversions done by the RecognizeKernelsPass are currently non-orthogonal. In particular, `kDropResultAndAliasArg0` probably won't work as intended if mixed with these (we probably need to promote kDropResultAndAliasArg0 to not be an arg-level thing anyway, as we have done with promoteTrailingOutTensor). This involved adding a new op `numpy.overwrite_array`. ``` numpy.overwrite_array %arg2 overwrites %arg0 : tensor<2x3xf32>, !numpy.ndarray<[2,3]:f32> ``` This models the destructive update behavior. Note that in the above op, we cannot simply RAUW %arg0 with a suitably conveted %arg2 (for example, %arg0 might have uses that are not dominated by %arg2, or might have an alias relation with some other array in the program). In general, we need a pass analogous to "SSA-formation" which knows how to see through these to uncover an underlying tensor program. Also, add tanh_out_e2e.py/div_inplace_e2e.py and fix some bitrot in refjit.py which is my running example I'm trying to get working.	2021-03-19 10:34:50 -07:00
Stella Laurenzo	046751254f	Refactor old tracing tests and remove deprecated ops. ... * Old doctests to run under lit. * Old custom filecheck tests -> pytest directory (under lit). * Rename some old ufunc ops in the tracer.	2020-06-29 16:19:03 -07:00
Stella Laurenzo	d3632af675	Add !numpy.any_dtype dialect type.	2020-04-29 18:20:42 -07:00
Stella Laurenzo	b4425fe1d2	Add numpy.ufunc_call op.	2020-04-29 17:49:56 -07:00
Stella Laurenzo	e845db8a20	Add builtin_ufunc and generic_ufunc ops.	2020-04-28 23:51:54 -07:00
Stella Laurenzo	953ef89a30	Add npcomp-opt and lit runner.	2020-04-26 17:55:15 -07:00