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

87fbc5e4 · feng_shuai · GitHub · 17879045 · 87fbc5e4
隐藏空白更改
内联并排

Showing with 2 addition and 1 deletion

paddle/fluid/inference/tensorrt/convert/multihead_matmul_op.cc ...e/fluid/inference/tensorrt/convert/multihead_matmul_op.cc +2 -1

未找到文件。
--- 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