Function TEST

test/cpp/inference/api/trt_disable_tensorrt_half_ops_test.cc:20–40

namespace inference {

TEST(TensorRT, disable_tensorrt_half_ops) {
  std::string model_dir = FLAGS_infer_model + "/resnet50";
  AnalysisConfig config;
  config.SetModel(model_dir);
  config.EnableUseGpu(100, 0);
  config.EnableTensorRtEngine(
      1 << 30, 1, 5, AnalysisConfig::Precision::kHalf, false, false);

  paddle_infer::experimental::InternalUtils::DisableTensorRtHalfOps(&config,
                                                                    {"conv2d"});

  std::vector<std::vector<PaddleTensor>> inputs_all;
  auto predictor = CreatePaddlePredictor(config);
  SetFakeImageInput(&inputs_all, model_dir, false, "__model__", "");

  std::vector<PaddleTensor> outputs;
  for (auto &input : inputs_all) {
    ASSERT_TRUE(predictor->Run(input, &outputs));
    predictor->ClearIntermediateTensor();
  }
}

}  // namespace inference
}  // namespace paddle