Generate librocsolver.jl with Clang.jl (#617)

* Generate librocsolver.jl with Clang.jl * Fix the issue with rocBLAS * Update highlevel.jl * [rocSOLVER] Update highlevel.jl
JuliaGPU · Sep 30, 2024 · 5407329 · 5407329
1 parent 78f5dec
commit 5407329
Show file tree

Hide file tree

Showing 17 changed files with 14,988 additions and 1,905 deletions.
diff --git a/gen/Project.toml b/gen/Project.toml
@@ -1,2 +1,3 @@
 [deps]
 Clang = "40e3b903-d033-50b4-a0cc-940c62c95e31"
+JuliaFormatter = "98e50ef6-434e-11e9-1051-2b60c6c9e899"
diff --git a/gen/hip/generator.jl b/gen/hip/generator.jl
@@ -5,6 +5,7 @@
 # #if defined(__HIP_PLATFORM_AMD__) && !defined(__HIP_PLATFORM_NVIDIA__)
 
 using Clang.Generators
+using JuliaFormatter
 
 include_dir = normpath("/opt/rocm/include")
 hip_dir = joinpath(include_dir, "hip")
@@ -22,3 +23,6 @@ headers = [
 
 ctx = create_context(headers, args, options)
 build!(ctx)
+
+path = options["general"]["output_file_path"]
+format_file(path, YASStyle())
diff --git a/gen/miopen/generator.jl b/gen/miopen/generator.jl
@@ -1,4 +1,5 @@
 using Clang.Generators
+using JuliaFormatter
 
 include_dir = normpath("/opt/rocm/include")
 miopen_dir = joinpath(include_dir, "miopen")
@@ -15,3 +16,6 @@ headers = [
 
 ctx = create_context(headers, args, options)
 build!(ctx)
+
+path = options["general"]["output_file_path"]
+format_file(path, YASStyle())
diff --git a/gen/miopen/miopen-generator.toml b/gen/miopen/miopen-generator.toml
@@ -4,3 +4,6 @@ library_name = "libMIOpen_path"
 output_file_path = "./libMIOpen.jl"
 jll_pkg_name = "MIOpen_jll"
 export_symbol_prefixes = []
+
+[codegen]
+use_ccall_macro = true
diff --git a/gen/rocblas/generator.jl b/gen/rocblas/generator.jl
@@ -1,4 +1,5 @@
 using Clang.Generators
+using JuliaFormatter
 
 include_dir = normpath("/opt/rocm/include")
 rocblas_dir = joinpath(include_dir, "rocblas")
@@ -15,3 +16,6 @@ headers = [
 
 ctx = create_context(headers, args, options)
 build!(ctx)
+
+path = options["general"]["output_file_path"]
+format_file(path, YASStyle())
diff --git a/gen/rocblas/rocblas-generator.toml b/gen/rocblas/rocblas-generator.toml
@@ -2,3 +2,6 @@
 library_name = "librocblas"
 output_file_path = "./librocblas.jl"
 export_symbol_prefixes = []
+
+[codegen]
+use_ccall_macro = true
diff --git a/gen/rocfft/generator.jl b/gen/rocfft/generator.jl
@@ -1,4 +1,5 @@
 using Clang.Generators
+using JuliaFormatter
 
 include_dir = normpath("/opt/rocm/include")
 rocfft_dir  = joinpath(include_dir, "rocfft")
@@ -15,3 +16,6 @@ headers = [
 
 ctx = create_context(headers, args, options)
 build!(ctx)
+
+path = options["general"]["output_file_path"]
+format_file(path, YASStyle())
diff --git a/gen/rocfft/rocfft-generator.toml b/gen/rocfft/rocfft-generator.toml
@@ -2,3 +2,6 @@
 library_name = "librocfft"
 output_file_path = "./librocfft.jl"
 export_symbol_prefixes = []
+
+[codegen]
+use_ccall_macro = true
diff --git a/gen/rocsolver/generator.jl b/gen/rocsolver/generator.jl
@@ -1,4 +1,8 @@
-using Clang.Generators
+# NOTE: for this to work, in opt/rocm/include/rocsolver/rocsolver-functions.h
+# replace all occurrences of "ROCSOLVER_EXPORT " by "".
+
+using Clang, Clang.Generators
+using JuliaFormatter
 
 include_dir = normpath("/opt/rocm/include")
 
@@ -15,4 +19,21 @@ headers = [
 ]
 
 ctx = create_context(headers, args, options)
-build!(ctx)
+build!(ctx, BUILDSTAGE_NO_PRINTING)
+
+# Only keep the wrapped headers because the dependencies are already wrapped with other headers.
+replace!(get_nodes(ctx.dag)) do node
+    path = Clang.get_filename(node.cursor)
+    should_wrap = any(headers) do header
+        occursin(header, path)
+    end
+    if !should_wrap
+        return ExprNode(node.id, Generators.Skip(), node.cursor, Expr[], node.adj)
+    end
+    return node
+end
+
+build!(ctx, BUILDSTAGE_PRINTING_ONLY)
+
+path = options["general"]["output_file_path"]
+format_file(path, YASStyle())
diff --git a/gen/rocsolver/rocsolver-generator.toml b/gen/rocsolver/rocsolver-generator.toml
@@ -2,3 +2,6 @@
 library_name = "librocsolver"
 output_file_path = "./librocsolver.jl"
 export_symbol_prefixes = []
+
+[codegen]
+use_ccall_macro = true
diff --git a/gen/rocsparse/generator.jl b/gen/rocsparse/generator.jl
@@ -1,4 +1,5 @@
 using Clang.Generators
+using JuliaFormatter
 
 include_dir = normpath("/opt/rocm/include")
 rocsparse_dir  = joinpath(include_dir, "rocsparse")
@@ -15,3 +16,6 @@ headers = [
 
 ctx = create_context(headers, args, options)
 build!(ctx)
+
+path = options["general"]["output_file_path"]
+format_file(path, YASStyle())
diff --git a/gen/rocsparse/rocsparse-generator.toml b/gen/rocsparse/rocsparse-generator.toml
@@ -2,3 +2,6 @@
 library_name = "librocsparse"
 output_file_path = "./librocsparse.jl"
 export_symbol_prefixes = []
+
+[codegen]
+use_ccall_macro = true