Commit f8173f07 authored by qinyiqun's avatar qinyiqun
Browse files

issue/87/fix: add metax-rt compile target

parent fd58c99c
......@@ -5,6 +5,8 @@ Handle::Handle(infiniDevice_t device, int device_id)
: InfiniopHandle{device, device_id},
_internal(std::make_shared<Handle::Internal>()) {}
Handle::Handle(int device_id) : Handle(INFINI_DEVICE_METAX, device_id) {}
auto Handle::internal() const -> const std::shared_ptr<Internal> & {
return _internal;
}
......
......@@ -6,6 +6,7 @@
namespace device::maca {
struct Handle : public InfiniopHandle {
Handle(int device_id);
class Internal;
auto internal() const -> const std::shared_ptr<Internal> &;
......
......@@ -60,7 +60,6 @@ infiniStatus_t Descriptor::calculate(
case INFINI_DTYPE_F32:
a_type = b_type = c_type = HPCC_R_32F;
compute_type = HCBLAS_COMPUTE_32F_FAST_TF32;
#endif
break;
default:
......
......@@ -132,6 +132,9 @@ target("infinirt")
if has_config("ascend-npu") then
add_deps("infinirt-ascend")
end
if has_config("metax-gpu") then
add_deps("infinirt-metax")
end
set_languages("cxx17")
set_installdir(os.getenv("INFINI_ROOT") or (os.getenv(is_host("windows") and "HOMEPATH" or "HOME") .. "/.infini"))
add_files("src/infinirt/*.cc")
......
......@@ -42,3 +42,13 @@ target("infiniop-metax")
add_files("../src/infiniop/ops/*/maca/*.maca", {rule = "maca"})
target_end()
target("infinirt-metax")
set_kind("static")
set_languages("cxx17")
on_install(function (target) end)
add_deps("infini-utils")
-- Add files
add_files("$(projectdir)/src/infinirt/maca/*.cc")
add_cxflags("-lstdc++ -Wall -Werror -fPIC")
target_end()
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