[Bug Fix] Fix AdaptivePool2d for ORT (#767)

fix adaptivepool2d
PaddlePaddle · Dec 1, 2022 · 11995e5 · 11995e5
1 parent 5a0eaa1
commit 11995e5
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/fastdeploy/backends/ort/ops/adaptive_pool2d.cc b/fastdeploy/backends/ort/ops/adaptive_pool2d.cc
@@ -76,7 +76,7 @@ void AdaptivePool2dKernel::Compute(OrtKernelContext* context) {
       context, 0, output_size_.data(), output_size_.size());
 
   float* output_data = ort_.GetTensorMutableData<float>(output);
-  if(this->provider_ == "CUDAExecutionProvider"){
+  if(!strcmp(this->provider_, "CUDAExecutionProvider")){
 #ifdef WITH_GPU
     auto compute_stream = ort_.KernelContext_GetGPUComputeStream(context);
     CudaAdaptivePool(input_size, output_size_, output_data, input_data, compute_stream, pooling_type_);
@@ -85,7 +85,7 @@ void AdaptivePool2dKernel::Compute(OrtKernelContext* context) {
             << "Will force to use CPU to run." << std::endl;
   CpuAdaptivePool(input_size, output_size_, input_data, output_data);
 #endif
-  }else{
+  } else {
     CpuAdaptivePool(input_size, output_size_, input_data, output_data);
   }
 }