Merge branch 'main' into scatter

xadupre · May 30, 2024 · 17db995 · 17db995
2 parents e244b1a + b60df02
commit 17db995
Showing 1 changed file with 6 additions and 5 deletions.
diff --git a/operators/cuda/negxplus1.h b/operators/cuda/negxplus1.h
@@ -4,29 +4,30 @@
 #pragma once
 #include "ocos.h"
 #include "negxplus1_impl.cuh"
+#include "ortx_common.h"
 
 namespace contrib {
 
 template <typename T>
 struct NegXPlus1 {
   template <typename TDict>
-  OrtStatusPtr OnModelAttach(const TDict& /*dict*/) {
-    return nullptr;
+  OrtxStatus OnModelAttach(const TDict& /*dict*/) {
+    return {};
   }
-  OrtStatusPtr Compute(Ort::Custom::CUDAKernelContext* ctx,
+  OrtxStatus Compute(Ort::Custom::CUDAKernelContext* ctx,
                        const ortc::Tensor<T>& input,
                        ortc::Tensor<T>& output) const {
     const T* input_data = input.Data();
     T* output_data = output.Allocate(input.Shape());
     auto input_length = input.NumberOfElement();
     if (0 == input_length) {
-      return nullptr;
+      return {};
     }
     LaunchNegXPlus1Kernel<T>(reinterpret_cast<cudaStream_t>(ctx->GetCudaStream()),
                              input_length,
                              input_data,
                              output_data);
-    return nullptr;
+    return {};
   }
 };