use transform_for_execution to get callable for torch compile

use transform_for_execution to get callable for torch compile #3675