Fix torch compile, script, export

Fix torch compile, script, export #693