qant_native_computing_toolkit_native.h

Go to the documentation of this file.

#ifndef QANT_NATIVE_COMPUTING_TOOLKIT_NATIVE_H
#define QANT_NATIVE_COMPUTING_TOOLKIT_NATIVE_H

 
#include <stdint.h>
 
#include <dlpack/dlpack.h>
 
#include "qant_native_computing_toolkit_generic.h"
 
#ifdef __cplusplus
namespace qant_native_computing_toolkit::native
{
    using ::qant_native_computing_toolkit::generic::QantMemoryHandle;
 
    extern "C"
    {
#endif

        DLManagedTensorVersioned *mul_npu(const uint32_t npu_id, DLManagedTensorVersioned const *us,
                                          DLManagedTensorVersioned const *vs);

        DLManagedTensorVersioned *calc_scaled_periodic_nl_fprop(const uint32_t npu_id, DLManagedTensorVersioned const *features, DLManagedTensorVersioned const *weights);

        DLManagedTensorVersioned *linear_fprop(const uint32_t npu_id, DLManagedTensorVersioned const *features,
                                               DLManagedTensorVersioned const *weights);

        DLManagedTensorVersioned *linear_fprop_preloaded_weights(DLManagedTensorVersioned const *features,
                                                                 QantMemoryHandle const *weights);

        int linear_fprop_preloaded(QantMemoryHandle const *features, QantMemoryHandle const *weights,
                                   QantMemoryHandle const *result);
 
#ifdef __cplusplus
    } // extern "C"
} // namespace qant_native_computing_toolkit::native
#endif
 
#endif // QANT_NATIVE_COMPUTING_TOOLKIT_NATIVE_H