From 7acb426f40c7f5274e54c7367641f5102366dc03 Mon Sep 17 00:00:00 2001 From: Thomas Viehmann Date: Thu, 10 Sep 2020 12:53:29 +0200 Subject: [PATCH] [ROCm] include mcpu and mtriple as target options --- src/target/target_kind.cc | 2 ++ 1 file changed, 2 insertions(+) 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))