Commit b96284ad authored by qinyiqun's avatar qinyiqun
Browse files

issue/87/musa: 跟随上游更新

parent e58e67a3
#ifndef __GEMM_MUSA_H__
#define __GEMM_MUSA_H__
#include "../gemm.h"
DESCRIPTOR(musa)
#endif // __GEMM_MUSA_H__
#include "../../../devices/musa/common_musa.h" #include "../../../devices/musa/common_musa.h"
#include "../../../devices/musa/musa_handle.h" #include "../../../devices/musa/musa_handle.h"
#include "matmul_musa.h" #include "gemm_musa.h"
namespace op::matmul::musa { namespace op::gemm::musa {
struct Descriptor::Opaque { struct Descriptor::Opaque {
std::shared_ptr<device::musa::Handle::Internal> internal; std::shared_ptr<device::musa::Handle::Internal> internal;
...@@ -139,4 +139,4 @@ infiniStatus_t Descriptor::calculate(void *workspace, ...@@ -139,4 +139,4 @@ infiniStatus_t Descriptor::calculate(void *workspace,
} }
} }
} // namespace op::matmul::musa } // namespace op::gemm::musa
...@@ -18,7 +18,7 @@ ...@@ -18,7 +18,7 @@
#include "maca/gemm_maca.h" #include "maca/gemm_maca.h"
#endif #endif
#ifdef ENABLE_MOORE_API #ifdef ENABLE_MOORE_API
#include "musa/matmul_musa.h" #include "musa/gemm_musa.h"
#endif #endif
#ifdef ENABLE_KUNLUN_API #ifdef ENABLE_KUNLUN_API
#include "kunlun/gemm_kunlun.h" #include "kunlun/gemm_kunlun.h"
......
#ifndef __MATMUL_MUSA_CUH__
#define __MATMUL_MUSA_CUH__
#include "../matmul.h"
DESCRIPTOR(musa)
#endif // __MATMUL_MUSA_CUH__
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment