Merge pull request #272 from InfiniTensor/issue/271

issue/271: xmake modify in moore gpu

xmake/musa.lua

-local MUSA_HOME = os.getenv("MUSA_INSTALL_PATH")
-add_includedirs(MUSA_HOME .. "/include")
-add_linkdirs(MUSA_HOME .. "/lib")
+local MUSA_ROOT = os.getenv("MUSA_ROOT") or os.getenv("MUSA_HOME") or os.getenv("MUSA_PATH")
+add_includedirs(MUSA_ROOT .. "/include")
+add_linkdirs(MUSA_ROOT .. "/lib")
 add_links("musa", "musart", "mudnn", "mublas")
 
 rule("mu")
@@ -13,13 +13,24 @@ rule("mu")
         local objectfile = target:objectfile(sourcefile)
         os.mkdir(path.directory(objectfile))
 
-        local mcc = MUSA_HOME .. "/bin/mcc"
+        local mcc = MUSA_ROOT .. "/bin/mcc"
         local includedirs = table.concat(target:get("includedirs"), " ")
-        local args = {"-c", sourcefile, "-o", objectfile, "-I" .. MUSA_HOME .. "/include", "-O3", "-fPIC", "-Wall", "-std=c++17", "-pthread"}
+
+        local args = {"-c", sourcefile, "-o", objectfile, "-I" .. MUSA_ROOT .. "/include", "-O3", "-fPIC", "-Wall", "-std=c++17", "-pthread"}
         for _, includedir in ipairs(target:get("includedirs")) do
             table.insert(args, "-I" .. includedir)
         end
 
+        -- ============================
+        -- Retrieve all preprocessor defines added to the current target
+        local defines = target:get("defines")
+        if defines then
+            for _, define in ipairs(defines) do
+                table.insert(args, "-D" .. define)
+            end
+        end
+        -- ===================================
+
         os.execv(mcc, args)
         table.insert(target:objectfiles(), objectfile)
     end)
@@ -31,7 +42,7 @@ target("infiniop-moore")
     set_languages("cxx17")
     set_warnings("all", "error")
     add_cxflags("-lstdc++", "-fPIC", "-Wno-comment")
-    add_files("../src/infiniop/devices/musa/*.cc", "../src/infiniop/ops/*/musa/*.cc")
+    add_files("../src/infiniop/devices/musa/*.cc")
     add_files("../src/infiniop/ops/*/musa/*.mu", {rule = "mu"})
 target_end()