"dockerfile/rocm5.1.x.dockerfile" did not exist on "e08b6d3a1cc32ac978182acd0be0ce197ca18e0d"
Unverified Commit 00c30a4a authored by PanZezhong1725's avatar PanZezhong1725 Committed by GitHub
Browse files

Merge branch 'main' into issue/240

parents 724d9692 d4b03cf7
#include "infinirt_musa.h"
#include "infinirt_moore.h"
#include "../../utils.h"
#include <musa_runtime.h>
#include <musa_runtime_api.h>
......
......@@ -608,6 +608,6 @@ def get_sync_func(device):
if device == InfiniDeviceEnum.CPU or device == InfiniDeviceEnum.CAMBRICON:
sync = None
else:
sync = getattr(torch, infiniDeviceEnum_str_map[device]).synchronize
sync = getattr(torch, torch_device_map[device]).synchronize
return sync
......@@ -119,7 +119,7 @@ option_end()
if has_config("moore-gpu") then
add_defines("ENABLE_MOORE_API")
includes("xmake/musa.lua")
includes("xmake/moore.lua")
end
-- 海光
......
......@@ -42,8 +42,8 @@ target("infiniop-moore")
set_languages("cxx17")
set_warnings("all", "error")
add_cxflags("-lstdc++", "-fPIC", "-Wno-comment")
add_files("../src/infiniop/devices/musa/*.cc")
add_files("../src/infiniop/ops/*/musa/*.mu", {rule = "mu"})
add_files("../src/infiniop/devices/moore/*.cc")
add_files("../src/infiniop/ops/*/moore/*.mu", {rule = "mu"})
target_end()
target("infinirt-moore")
......@@ -53,5 +53,5 @@ target("infinirt-moore")
add_deps("infini-utils")
set_warnings("all", "error")
add_cxflags("-lstdc++", "-fPIC")
add_files("../src/infinirt/musa/*.cc")
add_files("../src/infinirt/moore/*.cc")
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