mbs-octoml commented on code in PR #11770:
URL: https://github.com/apache/tvm/pull/11770#discussion_r907641605
##########
src/relay/backend/contrib/cutlass/codegen.cc:
##########
@@ -43,6 +43,16 @@ namespace cutlass {
namespace {
+/*! \brief Return the "cutlass" Target instance to use to guide compilation. */
+Target GetCutlassTarget() {
+ Target target = Target::Current(/*allow_not_defined=*/true);
+ if (!target.defined() || target->kind->name != "cutlass") {
+ // Use the default CUTLASS compilation options.
+ target = Target("cutlass");
Review Comment:
I'll beef up the comment. This is so folks who just want to use the default
CUTLASS compilation options (which are all very sensible) don't need to futz
with multi targets and can just do the usual target="cuda" or whatever
shorthand they prefer.
--
This is an automated message from the Apache Git Service.
To respond to the message, please log on to GitHub and use the
URL above to go to the specific comment.
To unsubscribe, e-mail: [email protected]
For queries about this service, please contact Infrastructure at:
[email protected]
