c_runtime_api.h

Go to the documentation of this file.

/*
 * Licensed to the Apache Software Foundation (ASF) under one
 * or more contributor license agreements.  See the NOTICE file
 * distributed with this work for additional information
 * regarding copyright ownership.  The ASF licenses this file
 * to you under the Apache License, Version 2.0 (the
 * "License"); you may not use this file except in compliance
 * with the License.  You may obtain a copy of the License at
 *
 *   http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing,
 * software distributed under the License is distributed on an
 * "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
 * KIND, either express or implied.  See the License for the
 * specific language governing permissions and limitations
 * under the License.
 */
 
/*
 * \file c_runtime_api.h
 * \brief MXNet runtime library.
 */
// Acknowledgement: This file originates from incubator-tvm
#ifndef MXNET_RUNTIME_C_RUNTIME_API_H_
#define MXNET_RUNTIME_C_RUNTIME_API_H_
 
#include <dlpack/dlpack.h>
 
#ifdef __cplusplus
extern "C" {
#endif
#include <mxnet/c_api.h>
#include <stdint.h>
#include <stddef.h>
 
typedef enum {
  // The type code of other types are compatible with DLPack.
  // The next few fields are extension types
  // that is used by MXNet API calls.
  kHandle        = 3U,
  kNull          = 4U,
  kMXNetType     = 5U,
  kMXNetContext  = 6U,
  kObjectHandle  = 7U,
  kStr           = 8U,
  kBytes         = 9U,
  kPyArg         = 10U,
  kNDArrayHandle = 11U,
  // Extension codes for other frameworks to integrate MXNet PackedFunc.
  // To make sure each framework's id do not conflict, use first and
  // last sections to mark ranges.
  // Open an issue at the repo if you need a section of code.
  kExtBegin  = 15U,
  kNNVMFirst = 16U,
  kNNVMLast  = 20U,
  // The following section of code is used for non-reserved types.
  kExtReserveEnd = 64U,
  kExtEnd        = 128U,
  // The rest of the space is used for custom, user-supplied datatypes
  kCustomBegin = 129U,
} MXNetTypeCode;
 
typedef union {
  int64_t v_int64;
  double v_float64;
  void* v_handle;
  const char* v_str;
  uint64_t v_uint64;
  DLDataType v_type;
} MXNetValue;
 
typedef struct {
  const char* data;
  size_t size;
} MXNetByteArray;
 
typedef void* MXNetFunctionHandle;
typedef void* MXNetObjectHandle;
 
MXNET_DLL int MXNetFuncFree(MXNetFunctionHandle func);
 
MXNET_DLL int MXNetFuncCall(MXNetFunctionHandle func,
                            MXNetValue* arg_values,
                            int* type_codes,
                            int num_args,
                            MXNetValue* ret_val,
                            int* ret_type_code);
 
MXNET_DLL int MXNetFuncGetGlobal(const char* name, MXNetFunctionHandle* out);
 
MXNET_DLL int MXNetFuncListGlobalNames(int* out_size, const char*** out_array);
 
MXNET_DLL int MXNetObjectFree(MXNetObjectHandle obj);
 
MXNET_DLL int MXNetObjectGetTypeIndex(MXNetObjectHandle obj, unsigned* out_tindex);
 
MXNET_DLL int MXNetObjectTypeKey2Index(const char* type_key, unsigned* out_tindex);
 
#ifdef __cplusplus
}  // extern "C"
#endif
#endif  // MXNET_RUNTIME_C_RUNTIME_API_H_