Profiler tensor shape to MojoOperator#359

Open

NASA1473 wants to merge 1 commit into

dev/m13_ilufrom

Commits on Jun 12, 2026

Wrap MojoOperator forward calls with torch.profiler.record_function so each op shows up in profiler traces with its class name, forward signature, and input tensor shape/dtype metadata. Adds helper…
NASA1473
committed