diff --git a/src/parsers.rs b/src/parsers.rs index be17b33..a9e582c 100644 --- a/src/parsers.rs +++ b/src/parsers.rs @@ -670,6 +670,9 @@ pub fn default_parsers<'t>( Box::new(SentinelFileParser::new("inductor_post_grad_graph", |e| { e.inductor_post_grad_graph.as_ref() })), + Box::new(SentinelFileParser::new("inductor_pre_grad_graph", |e| { + e.inductor_pre_grad_graph.as_ref() + })), Box::new(SentinelFileParser::new("dynamo_cpp_guards_str", |e| { e.dynamo_cpp_guards_str.as_ref() })), diff --git a/src/types.rs b/src/types.rs index ab474e0..5b476da 100644 --- a/src/types.rs +++ b/src/types.rs @@ -446,6 +446,7 @@ pub struct Envelope { pub aot_backward_graph: Option, pub aot_inference_graph: Option, pub aot_joint_graph: Option, + pub inductor_pre_grad_graph: Option, pub inductor_post_grad_graph: Option, pub dynamo_cpp_guards_str: Option, pub inductor_output_code: Option,