Use of OrtxStatus in kernel NegXPlus1 (microsoft#732)

* first draft for NegXPlus1 * complete * fix unit test * rename one test * remove test if not cuda * switch to OrtxStatus --------- Co-authored-by: Wenbing Li <[email protected]>
xadupre · May 30, 2024 · b60df02 · b60df02
1 parent 95a49fa
commit b60df02
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 {};
   }
 };