mirror of
https://github.com/Gericom/teak-llvm.git
synced 2025-06-25 14:28:54 -04:00
9767089d00
This patch renames -f{no-}cuda-rdc to -f{no-}gpu-rdc and keeps the original
options as aliases. When -fgpu-rdc is off,
clang will assume the device code in each translation unit does not call
external functions except those in the device library, therefore it is possible
to compile the device code in each translation unit to self-contained kernels
and embed them in the host object, so that the host object behaves like
usual host object which can be linked by lld.
The benefits of this feature is: 1. allow users to create static libraries which
can be linked by host linker; 2. amortized device code linking time.
This patch modifies HIP action builder to insert actions for linking device
code and generating HIP fatbin, and pass HIP fatbin to host backend action.
It extracts code for constructing command for generating HIP fatbin as
a function so that it can be reused by early finalization. It also modifies
codegen of HIP host constructor functions to embed the device fatbin
when it is available.
Differential Revision: https://reviews.llvm.org/D52377
llvm-svn: 343611
148 lines
6.3 KiB
Plaintext
148 lines
6.3 KiB
Plaintext
// Tests that ptxas and fatbinary are invoked correctly during CUDA
|
|
// compilation.
|
|
//
|
|
// REQUIRES: clang-driver
|
|
// REQUIRES: x86-registered-target
|
|
// REQUIRES: nvptx-registered-target
|
|
|
|
// Regular compiles with -O{0,1,2,3,4,fast}. -O4 and -Ofast map to ptxas O3.
|
|
// RUN: %clang -### -target x86_64-linux-gnu -O0 -c %s 2>&1 \
|
|
// RUN: | FileCheck -check-prefixes=CHECK,ARCH64,SM20,OPT0 %s
|
|
// RUN: %clang -### -target x86_64-linux-gnu -O1 -c %s 2>&1 \
|
|
// RUN: | FileCheck -check-prefixes=CHECK,ARCH64,SM20,OPT1 %s
|
|
// RUN: %clang -### -target x86_64-linux-gnu -O2 -c %s 2>&1 \
|
|
// RUN: | FileCheck -check-prefixes=CHECK,ARCH64,SM20,OPT2 %s
|
|
// RUN: %clang -### -target x86_64-linux-gnu -O3 -c %s 2>&1 \
|
|
// RUN: | FileCheck -check-prefixes=CHECK,ARCH64,SM20,OPT3 %s
|
|
// RUN: %clang -### -target x86_64-linux-gnu -O4 -c %s 2>&1 \
|
|
// RUN: | FileCheck -check-prefixes=CHECK,ARCH64,SM20,OPT3 %s
|
|
// RUN: %clang -### -target x86_64-linux-gnu -Ofast -c %s 2>&1 \
|
|
// RUN: | FileCheck -check-prefixes=CHECK,ARCH64,SM20,OPT3 %s
|
|
// Generating relocatable device code
|
|
// RUN: %clang -### -target x86_64-linux-gnu -fgpu-rdc -c %s 2>&1 \
|
|
// RUN: | FileCheck -check-prefixes=CHECK,ARCH64,SM20,RDC %s
|
|
|
|
// With debugging enabled, ptxas should be run with with no ptxas optimizations.
|
|
// RUN: %clang -### -target x86_64-linux-gnu --cuda-noopt-device-debug -O2 -g -c %s 2>&1 \
|
|
// RUN: | FileCheck -check-prefixes=CHECK,ARCH64,SM20,DBG %s
|
|
|
|
// --no-cuda-noopt-device-debug overrides --cuda-noopt-device-debug.
|
|
// RUN: %clang -### -target x86_64-linux-gnu --cuda-noopt-device-debug \
|
|
// RUN: --no-cuda-noopt-device-debug -O2 -c %s 2>&1 \
|
|
// RUN: | FileCheck -check-prefixes=CHECK,ARCH64,SM20,OPT2 %s
|
|
|
|
// Regular compile without -O. This should result in us passing -O0 to ptxas.
|
|
// RUN: %clang -### -target x86_64-linux-gnu -c %s 2>&1 \
|
|
// RUN: | FileCheck -check-prefixes=CHECK,ARCH64,SM20,OPT0 %s
|
|
|
|
// Regular compiles with -Os and -Oz. For lack of a better option, we map
|
|
// these to ptxas -O3.
|
|
// RUN: %clang -### -target x86_64-linux-gnu -Os -c %s 2>&1 \
|
|
// RUN: | FileCheck -check-prefixes=CHECK,ARCH64,SM20,OPT2 %s
|
|
// RUN: %clang -### -target x86_64-linux-gnu -Oz -c %s 2>&1 \
|
|
// RUN: | FileCheck -check-prefixes=CHECK,ARCH64,SM20,OPT2 %s
|
|
|
|
// Regular compile targeting sm_35.
|
|
// RUN: %clang -### -target x86_64-linux-gnu --cuda-gpu-arch=sm_35 -c %s 2>&1 \
|
|
// RUN: | FileCheck -check-prefixes=CHECK,ARCH64,SM35 %s
|
|
// Separate compilation targeting sm_35.
|
|
// RUN: %clang -### -target x86_64-linux-gnu --cuda-gpu-arch=sm_35 -fgpu-rdc -c %s 2>&1 \
|
|
// RUN: | FileCheck -check-prefixes=CHECK,ARCH64,SM35,RDC %s
|
|
|
|
// 32-bit compile.
|
|
// RUN: %clang -### -target i386-linux-gnu -c %s 2>&1 \
|
|
// RUN: | FileCheck -check-prefixes=CHECK,ARCH32,SM20 %s
|
|
// 32-bit compile when generating relocatable device code.
|
|
// RUN: %clang -### -target i386-linux-gnu -fgpu-rdc -c %s 2>&1 \
|
|
// RUN: | FileCheck -check-prefixes=CHECK,ARCH32,SM20,RDC %s
|
|
|
|
// Compile with -fintegrated-as. This should still cause us to invoke ptxas.
|
|
// RUN: %clang -### -target x86_64-linux-gnu -fintegrated-as -c %s 2>&1 \
|
|
// RUN: | FileCheck -check-prefixes=CHECK,ARCH64,SM20,OPT0 %s
|
|
// Check that we still pass -c when generating relocatable device code.
|
|
// RUN: %clang -### -target x86_64-linux-gnu -fintegrated-as -fgpu-rdc -c %s 2>&1 \
|
|
// RUN: | FileCheck -check-prefixes=CHECK,ARCH64,SM20,RDC %s
|
|
|
|
// Check -Xcuda-ptxas and -Xcuda-fatbinary
|
|
// RUN: %clang -### -target x86_64-linux-gnu -c -Xcuda-ptxas -foo1 \
|
|
// RUN: -Xcuda-fatbinary -bar1 -Xcuda-ptxas -foo2 -Xcuda-fatbinary -bar2 %s 2>&1 \
|
|
// RUN: | FileCheck -check-prefixes=CHECK,SM20,PTXAS-EXTRA,FATBINARY-EXTRA %s
|
|
|
|
// MacOS spot-checks
|
|
// RUN: %clang -### -target x86_64-apple-macosx -O0 -c %s 2>&1 \
|
|
// RUN: | FileCheck -check-prefixes=CHECK,ARCH64,SM20,OPT0 %s
|
|
// RUN: %clang -### -target x86_64-apple-macosx --cuda-gpu-arch=sm_35 -c %s 2>&1 \
|
|
// RUN: | FileCheck -check-prefixes=CHECK,ARCH64,SM35 %s
|
|
// RUN: %clang -### -target i386-apple-macosx -c %s 2>&1 \
|
|
// RUN: | FileCheck -check-prefixes=CHECK,ARCH32,SM20 %s
|
|
|
|
// Check relocatable device code generation on MacOS.
|
|
// RUN: %clang -### -target x86_64-apple-macosx -O0 -fgpu-rdc -c %s 2>&1 \
|
|
// RUN: | FileCheck -check-prefixes=CHECK,ARCH64,SM20,RDC %s
|
|
// RUN: %clang -### -target x86_64-apple-macosx --cuda-gpu-arch=sm_35 -fgpu-rdc -c %s 2>&1 \
|
|
// RUN: | FileCheck -check-prefixes=CHECK,ARCH64,SM35,RDC %s
|
|
// RUN: %clang -### -target i386-apple-macosx -fgpu-rdc -c %s 2>&1 \
|
|
// RUN: | FileCheck -check-prefixes=CHECK,ARCH32,SM20,RDC %s
|
|
|
|
// Check that CLANG forwards the -v flag to PTXAS.
|
|
// RUN: %clang -### -save-temps -no-canonical-prefixes -v %s 2>&1 \
|
|
// RUN: | FileCheck -check-prefix=CHK-PTXAS-VERBOSE %s
|
|
|
|
// Match clang job that produces PTX assembly.
|
|
// CHECK: "-cc1"
|
|
// ARCH64-SAME: "-triple" "nvptx64-nvidia-cuda"
|
|
// ARCH32-SAME: "-triple" "nvptx-nvidia-cuda"
|
|
// RDC-SAME: "-fgpu-rdc"
|
|
// CHECK-NOT: "-fgpu-rdc"
|
|
// SM20-SAME: "-target-cpu" "sm_20"
|
|
// SM35-SAME: "-target-cpu" "sm_35"
|
|
// SM20-SAME: "-o" "[[PTXFILE:[^"]*]]"
|
|
// SM35-SAME: "-o" "[[PTXFILE:[^"]*]]"
|
|
|
|
// Match the call to ptxas (which assembles PTX to SASS).
|
|
// CHECK: ptxas
|
|
// ARCH64-SAME: "-m64"
|
|
// ARCH32-SAME: "-m32"
|
|
// OPT0-SAME: "-O0"
|
|
// OPT0-NOT: "-g"
|
|
// OPT1-SAME: "-O1"
|
|
// OPT1-NOT: "-g"
|
|
// OPT2-SAME: "-O2"
|
|
// OPT2-NOT: "-g"
|
|
// OPT3-SAME: "-O3"
|
|
// OPT3-NOT: "-g"
|
|
// DBG-SAME: "-g" "--dont-merge-basicblocks" "--return-at-end"
|
|
// SM20-SAME: "--gpu-name" "sm_20"
|
|
// SM35-SAME: "--gpu-name" "sm_35"
|
|
// SM20-SAME: "--output-file" "[[CUBINFILE:[^"]*]]"
|
|
// SM35-SAME: "--output-file" "[[CUBINFILE:[^"]*]]"
|
|
// CHECK-SAME: "[[PTXFILE]]"
|
|
// PTXAS-EXTRA-SAME: "-foo1"
|
|
// PTXAS-EXTRA-SAME: "-foo2"
|
|
// RDC-SAME: "-c"
|
|
// CHECK-NOT: "-c"
|
|
|
|
// Match the call to fatbinary (which combines all our PTX and SASS into one
|
|
// blob).
|
|
// CHECK: fatbinary
|
|
// CHECK-SAME-DAG: "--cuda"
|
|
// ARCH64-SAME-DAG: "-64"
|
|
// ARCH32-SAME-DAG: "-32"
|
|
// CHECK-DAG: "--create" "[[FATBINARY:[^"]*]]"
|
|
// SM20-SAME-DAG: "--image=profile=compute_20,file=[[PTXFILE]]"
|
|
// SM35-SAME-DAG: "--image=profile=compute_35,file=[[PTXFILE]]"
|
|
// SM20-SAME-DAG: "--image=profile=sm_20,file=[[CUBINFILE]]"
|
|
// SM35-SAME-DAG: "--image=profile=sm_35,file=[[CUBINFILE]]"
|
|
// FATBINARY-EXTRA-SAME: "-bar1"
|
|
// FATBINARY-EXTRA-SAME: "-bar2"
|
|
|
|
// Match the clang job for host compilation.
|
|
// CHECK: "-cc1"
|
|
// ARCH64-SAME: "-triple" "x86_64-
|
|
// ARCH32-SAME: "-triple" "i386-
|
|
// CHECK-SAME: "-fcuda-include-gpubinary" "[[FATBINARY]]"
|
|
// RDC-SAME: "-fgpu-rdc"
|
|
// CHECK-NOT: "-fgpu-rdc"
|
|
|
|
// CHK-PTXAS-VERBOSE: ptxas{{.*}}" "-v"
|