search

iree-org / iree-turbine

IREE's PyTorch Frontend, based on Torch Dynamo.
Apache License 2.0
55 stars 25 forks source link

Add tensor tracing in eager mode #251

Closed sogartar closed 1 week ago

sogartar commented 1 week ago

We don't trace in eager mode in ops.iree.trace_tensor.

This adds the ability to set a tensor trace callback with a default implementation that records the tensor to a npy file.