torch.jit already allows a mix of tracing and AOT via combining torch.jit.trace
, torch.jit.script
and torch.jit.script_if_tracing
. Will there be a path for something like torch.jit.trace
to produce the new primitive ops rather than ATen ops?
@_sean_silva was there a follow-up to this discussion in Slack somewhere? If so could you please post a link or a summary?