kernel.hpp

Go to the documentation of this file.

// SPDX-License-Identifier: Apache-2.0
// Copyright 2024 XCENA Inc.
 
#pragma once
 
#include "mu/crt0/call_main.hpp"
 
namespace mu
{
typedef struct
{
    const char* name;
    void (*func)(unsigned int mode, unsigned int ndarrayValueBit, char* argv);
} KernelInfo;
}  // namespace mu
 
#ifndef MU_NON_BLOCKING_KERNEL_INIT
#define MU_NON_BLOCKING_KERNEL_INIT(FUNC) \
    MU_KERNEL_ADD(FUNC)                   \
    extern "C" [[gnu::used]] const uint64_t __mu_kernel_type__ = 1;
#endif
 
// for safe ##
#ifndef _MU_CONCAT
#define _MU_CONCAT(a, b) a##b
#endif
#ifndef MU_CONCAT
#define MU_CONCAT(a, b) _MU_CONCAT(a, b)
#endif
 
#ifndef MU_KERNEL_DEFINE
#define MU_KERNEL_DEFINE(TARGET, ID)                                                                         \
    extern "C" void MU_CONCAT(__mu_kernel_, ID)(unsigned int mode, unsigned int ndarrayValueBit, char* argv) \
    {                                                                                                        \
        mu::callMain(TARGET, mode, ndarrayValueBit, argv);                                                   \
    }                                                                                                        \
    extern "C" [[gnu::used]] const mu::KernelInfo MU_CONCAT(__mu_func_, ID) = {                              \
        #ID,                                                                                                 \
        MU_CONCAT(__mu_kernel_, ID)};
#endif
 
#ifndef MU_KERNEL_ADD
#define MU_KERNEL_ADD(FUNC) MU_KERNEL_DEFINE(FUNC, FUNC)
#endif
 
#ifndef __pxl_kernel__
#define __pxl_kernel__ [[clang::annotate("mu::offload")]]
#endif
 
#ifndef __mu_shared__
#define __mu_shared__ [[clang::annotate("mu::share")]]
#endif
 
#ifndef __mu_device__
#define __mu_device__ [[clang::annotate("mu::device")]]
#endif