torch-mlir/python/npcomp/tracing
Stella Laurenzo a5f755d406 Implement __array_func__ hook and use it to trace np.dot.
* Creates an abstraction/registry around emitters (intended to generalize to AST compilation as well).
* Reworks ufuncs to use the same mechanism as array funcs.
* Adds the numpy.dot op.
2020-05-04 15:47:01 -07:00
..
__init__.py Cleanup python namespace a bit for standalone use. 2020-05-02 21:54:13 -07:00
context.py Add hook for __array_function__ and (failing) np.dot sample. 2020-05-03 13:39:30 -07:00
emitters.py Implement __array_func__ hook and use it to trace np.dot. 2020-05-04 15:47:01 -07:00
mlir_trace.py Implement __array_func__ hook and use it to trace np.dot. 2020-05-04 15:47:01 -07:00
mlir_trace_test.py Rework the poc tracer to be structured how intended. 2020-05-02 19:52:21 -07:00