Infer if current compiler is clang

nervosnetwork · Aug 31, 2023 · a6c1445 · a6c1445
1 parent 3556380
commit a6c1445
Showing 1 changed file with 7 additions and 2 deletions.
diff --git a/build.rs b/build.rs
@@ -33,17 +33,22 @@ fn main() {
                 None => "clang-16".into(),
             };
 
+            build.compiler(clang);
+        }
+
+        let compiler = build.get_compiler();
+        if compiler.is_like_clang() {
             build
-                .compiler(clang)
                 .no_default_flags(true)
                 .flag("--target=riscv64")
                 .flag("-march=rv64imc_zba_zbb_zbc_zbs");
-        } else {
+        } else if compiler.is_like_gnu() {
             build
                 .flag("-nostartfiles")
                 .flag("-Wno-dangling-pointer")
                 .flag("-Wno-nonnull-compare");
         }
+
         build.compile("dl-c-impl");
     }