diff --git a/src/target/target_kind.cc b/src/target/target_kind.cc index 29f16925968d..99c1f9dda63d 100644 --- a/src/target/target_kind.cc +++ b/src/target/target_kind.cc @@ -114,6 +114,8 @@ TVM_REGISTER_TARGET_KIND("rocm") .add_attr_option>("keys") .add_attr_option>("libs") .add_attr_option("device") + .add_attr_option("mcpu") + .add_attr_option("mtriple") .add_attr_option("model") .add_attr_option("system-lib") .add_attr_option("max_num_threads", Integer(256))