fix: oss just support sm>=75 (PaddlePaddle#48731)

lxsbupt · Dec 7, 2022 · 81c9bc4 · 81c9bc4
1 parent ef765ea
commit 81c9bc4
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/paddle/fluid/inference/tensorrt/convert/multihead_matmul_op.cc b/paddle/fluid/inference/tensorrt/convert/multihead_matmul_op.cc
@@ -400,7 +400,8 @@ class MultiheadMatMulOpConverter : public OpConverter {
         }
       } else {
         if (input_dims.d[1] <= 384 && !bias_qk_attr &&
-            engine_->precision() != AnalysisConfig::Precision::kFloat32) {
+            engine_->precision() != AnalysisConfig::Precision::kFloat32 &&
+            platform::GetGPUComputeCapability(0) >= 75) {
           /*
             * input_dims.d[0]: batch(-1)
             * input_dims.d[1]: length:256