Skip to content

Commit 140c37d

Browse files
Wolfram70grypp
andcommitted
Apply suggestions from code review
Co-authored-by: Guray Ozen <[email protected]>
1 parent ef6abe5 commit 140c37d

File tree

1 file changed

+4
-0
lines changed

1 file changed

+4
-0
lines changed

mlir/include/mlir/Dialect/LLVMIR/NVVMRequiresSMTraits.td

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -27,6 +27,8 @@ def RequiresSMInterface: OpInterface<"RequiresSMInterface"> {
2727
];
2828
}
2929

30+
// OP requires a specified minimum SM value or higher;
31+
// it is not architecture-specific.
3032
class NVVMRequiresSM<int minVersion> :
3133
ParamNativeOpTrait<"NVVMRequiresSM", !cast<string>(minVersion)>;
3234

@@ -35,6 +37,8 @@ class StrJoin<string sep, list<string> str_list> {
3537
!if(!eq(a, ""), b, !if(!eq(b, ""), a, !strconcat(a, sep, b))));
3638
}
3739

40+
// OP requires an exact SM match along with
41+
// architecture acceleration.
3842
class NVVMRequiresSMa<list<int> smVersions> :
3943
ParamNativeOpTrait<"NVVMRequiresSMa",
4044
StrJoin<",", !foreach(vers, smVersions,

0 commit comments

Comments
 (0)