issue/89 增加沐曦运行时

src/infinirt/maca/infinirt_maca.cc

+#include "infinirt_maca.h"
+#include "../../utils.h"
+#include <hcr/hc_runtime.h>
+#include <hcr/hc_runtime_api.h>
+
+#define CHECK_MACART(RT_API) CHECK_INTERNAL(RT_API, hcSuccess)
+
+namespace infinirt::cuda {
+infiniStatus_t getDeviceCount(int *count) {
+    CHECK_MACART(hcGetDeviceCount(count));
+    return INFINI_STATUS_SUCCESS;
+}
+
+infiniStatus_t setDevice(int device_id) {
+    CHECK_MACART(hcSetDevice(device_id));
+    return INFINI_STATUS_SUCCESS;
+}
+
+infiniStatus_t deviceSynchronize() {
+    CHECK_MACART(hcDeviceSynchronize());
+    return INFINI_STATUS_SUCCESS;
+}
+
+infiniStatus_t streamCreate(infinirtStream_t *stream_ptr) {
+    hcStream_t stream;
+    CHECK_MACART(hcStreamCreate(&stream));
+    *stream_ptr = stream;
+    return INFINI_STATUS_SUCCESS;
+}
+
+infiniStatus_t streamDestroy(infinirtStream_t stream) {
+    CHECK_MACART(hcStreamDestroy((hcStream_t)stream));
+    return INFINI_STATUS_SUCCESS;
+}
+
+infiniStatus_t streamSynchronize(infinirtStream_t stream) {
+    CHECK_MACART(hcStreamSynchronize((hcStream_t)stream));
+    return INFINI_STATUS_SUCCESS;
+}
+
+infiniStatus_t streamWaitEvent(infinirtStream_t stream, infinirtEvent_t event) {
+    CHECK_MACART(hcStreamWaitEvent((hcStream_t)stream, (hcEvent_t)event));
+    return INFINI_STATUS_SUCCESS;
+}
+
+infiniStatus_t eventCreate(infinirtEvent_t *event_ptr) {
+    hcEvent_t event;
+    CHECK_MACART(hcEventCreate(&event));
+    *event_ptr = event;
+    return INFINI_STATUS_SUCCESS;
+}
+
+infiniStatus_t eventRecord(infinirtEvent_t event, infinirtStream_t stream) {
+    CHECK_MACART(hcEventRecord((hcEvent_t)event, (hcStream_t)stream));
+    return INFINI_STATUS_SUCCESS;
+}
+
+infiniStatus_t eventQuery(infinirtEvent_t event, infinirtEventStatus_t *status_ptr) {
+    CHECK_MACART(hcEventQuery((hcEvent_t)event));
+    return INFINI_STATUS_SUCCESS;
+}
+
+infiniStatus_t eventSynchronize(infinirtEvent_t event) {
+    CHECK_MACART(hcEventSynchronize((hcEvent_t)event));
+    return INFINI_STATUS_SUCCESS;
+}
+
+infiniStatus_t eventDestroy(infinirtEvent_t event) {
+    CHECK_MACART(hcEventDestroy((hcEvent_t)event));
+    return INFINI_STATUS_SUCCESS;
+}
+
+infiniStatus_t mallocDevice(void **p_ptr, size_t size) {
+    CHECK_MACART(hcMalloc(p_ptr, size));
+    return INFINI_STATUS_SUCCESS;
+}
+
+infiniStatus_t mallocHost(void **p_ptr, size_t size) {
+    CHECK_MACART(hcMallocHost(p_ptr, size));
+    return INFINI_STATUS_SUCCESS;
+}
+
+infiniStatus_t freeDevice(void *ptr) {
+    CHECK_MACART(hcFree(ptr));
+    return INFINI_STATUS_SUCCESS;
+}
+
+infiniStatus_t freeHost(void *ptr) {
+    CHECK_MACART(hcFreeHost(ptr));
+    return INFINI_STATUS_SUCCESS;
+}
+
+cudaMemcpyKind toMacaMemcpyKind(infinirtMemcpyKind_t kind) {
+    switch (kind) {
+    case INFINIRT_MEMCPY_H2D:
+        return hcMemcpyHostToDevice;
+    case INFINIRT_MEMCPY_D2H:
+        return hcMemcpyDeviceToHost;
+    case INFINIRT_MEMCPY_D2D:
+        return hcMemcpyDeviceToDevice;
+    case INFINIRT_MEMCPY_H2H:
+        return hcMemcpyHostToHost;
+    default:
+        return hcMemcpyDefault;
+    }
+}
+
+infiniStatus_t memcpy(void *dst, const void *src, size_t size, infinirtMemcpyKind_t kind) {
+    CHECK_MACART(hcMemcpy(dst, src, size, toMacaMemcpyKind(kind)));
+    return INFINI_STATUS_SUCCESS;
+}
+
+infiniStatus_t memcpyAsync(void *dst, const void *src, size_t size, infinirtMemcpyKind_t kind, infinirtStream_t stream) {
+    CHECK_MACART(hcMemcpyAsync(dst, src, size, toMacaMemcpyKind(kind), (hcStream_t)stream));
+    return INFINI_STATUS_SUCCESS;
+}
+
+infiniStatus_t mallocAsync(void **p_ptr, size_t size, infinirtStream_t stream) {
+    CHECK_MACART(hcMallocAsync(p_ptr, size, (hcStream_t)stream));
+    return INFINI_STATUS_SUCCESS;
+}
+
+infiniStatus_t freeAsync(void *ptr, infinirtStream_t stream) {
+    CHECK_MACART(hcFreeAsync(ptr, (hcStream_t)stream));
+    return INFINI_STATUS_SUCCESS;
+}
+} // namespace infinirt::cuda

src/infinirt/maca/infinirt_maca.h

+#ifndef __INFINIRT_MACA_H__
+#define __INFINIRT_MACA_H__
+#include "../infinirt_impl.h"
+
+namespace infinirt::mca {
+#ifdef ENABLE_MACA_API
+INFINIRT_DEVICE_API_IMPL
+#else
+INFINIRT_DEVICE_API_NOOP
+#endif
+} // namespace infinirt::mca
+
+#endif // __INFINIRT_MACA_H__

xmake/maca.lua

@@ -34,8 +34,11 @@ rule_end()
 target("infiniop-metax")
     set_kind("static")
     on_install(function (target) end)
+    add_cxflags("-lstdc++ -Wall -fPIC")
     set_languages("cxx17")
+    set_warnings("all")
+
     add_files("../src/infiniop/devices/maca/*.cc", "../src/infiniop/ops/*/maca/*.cc")
     add_files("../src/infiniop/ops/*/maca/*.maca", {rule = "maca"})
-    add_cxflags("-lstdc++ -Werror -fPIC")
+
 target_end()