Skip to content

Commit 87cd10e

Browse files
authored
ProbProg: Pure StableHLO Migration (#2692)
* clean up * simulate/generate/regenerate * minor * simulate/generate/regenerate ODS * vendor new tensor ops * simulate/generate/regenerate clone update * mh * new mh ODS * remove lower update pattern register * migrate test * lower MCMC pattern migrate * clean up * mcmc migrate * bug fix * minor * test * fix up * fix
1 parent aec862f commit 87cd10e

26 files changed

+2612
-2632
lines changed

enzyme/Enzyme/MLIR/Dialect/EnzymeEnums.td

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -120,7 +120,7 @@ def NUTSConfigAttr : Enzyme_Attr<"NUTSConfig", "nuts_config"> {
120120
DefaultValuedParameter<"bool", "true">:$adapt_step_size,
121121
DefaultValuedParameter<"bool", "true">:$adapt_mass_matrix
122122
);
123-
let assemblyFormat = "`<` `max_tree_depth` `=` $max_tree_depth (`,` `max_delta_energy` `=` $max_delta_energy^)? (`,` `adapt_step_size` `=` $adapt_step_size^)? (`,` `adapt_mass_matrix` `=` $adapt_mass_matrix^)? `>`";
123+
let assemblyFormat = "`<` (`max_tree_depth` `=` $max_tree_depth^)? (`,` `max_delta_energy` `=` $max_delta_energy^)? (`,` `adapt_step_size` `=` $adapt_step_size^)? (`,` `adapt_mass_matrix` `=` $adapt_mass_matrix^)? `>`";
124124
}
125125

126126
#endif // ENZYME_ENUMS

0 commit comments

Comments
 (0)