Let hipcc handle gfx906 for hip-clang

2018-10-21 10:03:33 -04:00
parent 4a00b244a3
commit 283fb42604
1 changed files with 5 additions and 1 deletions
@@ -687,7 +687,11 @@ if($HIP_PLATFORM eq "hcc" or $HIP_PLATFORM eq "clang"){
        $HIPCXXFLAGS .= " -D__HIP_ARCH_GFX900__=1 ";
    }
    if ($target_gfx906 eq 1) {
-        $HIPLDFLAGS .= " --amdgpu-target=gfx906";
+        $GPU_ARCH_ARG = $GPU_ARCH_OPT . "gfx906";
+        $HIPLDFLAGS .= $GPU_ARCH_ARG;
+        if ($HIP_PLATFORM eq 'clang') {
+            $HIPCXXFLAGS .= $GPU_ARCH_ARG;;
+        }
        $HIPCXXFLAGS .= " -D__HIP_ARCH_GFX906__=1 ";
    }
 }