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