[Fix](bangc-ops):fix core context code (#334)

Co-authored-by: tudejiang <[email protected]>
Cambricon · Dec 15, 2022 · d462968 · d462968
1 parent feb826a
commit d462968
Showing 1 changed file with 9 additions and 0 deletions.
diff --git a/bangc-ops/core/context.cpp b/bangc-ops/core/context.cpp
@@ -233,6 +233,15 @@ mluOpStatus_t mluOpUpdateContextInformation(mluOpHandle_t handle) {
 }
 
 mluOpStatus_t mluOpSetAtomicsMode(mluOpHandle_t handle,
+                                  mluOpAtomicsMode_t atomics_mode) {
+  PARAM_CHECK("[mluOpSetAtomicsMode]", handle != NULL);
+
+  handle->atomics_mode = atomics_mode;
+
+  return MLUOP_STATUS_SUCCESS;
+}
+
+mluOpStatus_t mluOpGetAtomicsMode(mluOpHandle_t handle,
                                   mluOpAtomicsMode_t *atomics_mode) {
   PARAM_CHECK("[mluOpGetAtomicsMode]", handle != NULL);
   PARAM_CHECK("[mluOpGetAtomicsMode]", atomics_mode != NULL);