"models/vision/vscode:/vscode.git/clone" did not exist on "2e9910bdb703da3ebeb295d06651648dc100eaa1"
Unverified Commit 07f62104 authored by Shawn Liu's avatar Shawn Liu Committed by GitHub
Browse files

Modify the SM architecture number to support Thor’s sm110. (#957)

parent f6d4bd3a
......@@ -57,7 +57,7 @@ bool TargetIsSm100(Target target) {
if (!TargetIsCuda(target))
return false;
int arch = GetArchInt(target);
return arch >= 100 & arch <= 103;
return arch >= 100 & arch <= 110;
}
bool TargetIsSM120(Target target) {
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment