clang/test/CodeGenHLSL/semantics/DispatchThreadID.hlsl

   1 // RUN: %clang_cc1 -triple dxil-pc-shadermodel6.3-library -x hlsl -emit-llvm -finclude-default-header -disable-llvm-passes -o - %s | FileCheck %s --check-prefixes=CHECK,CHECK-DXIL -DTARGET=dx
   2 // RUN: %clang_cc1 -triple spirv-linux-vulkan-library -x hlsl -emit-llvm -finclude-default-header -disable-llvm-passes -o - %s | FileCheck %s --check-prefixes=CHECK,CHECK-SPIRV -DTARGET=spv
   3
   4 // Make sure SV_DispatchThreadID translated into dx.thread.id.
   5
   6 // CHECK:       define void @foo()
   7 // CHECK-DXIL:  %[[#ID:]] = call i32 @llvm.[[TARGET]].thread.id(i32 0)
   8 // CHECK-SPIRV: %[[#ID:]] = call i32 @llvm.[[TARGET]].thread.id(i32 0)
   9 // CHECK-DXIL:       call void @{{.*}}foo{{.*}}(i32 %[[#ID]])
  10 // CHECK-SPIRV:      call spir_func void @{{.*}}foo{{.*}}(i32 %[[#ID]])
  11 [shader("compute")]
  12 [numthreads(8,8,1)]
  13 void foo(uint Idx : SV_DispatchThreadID) {}
  14
  15 // CHECK:       define void @bar()
  16 // CHECK:       %[[#ID_X:]] = call i32 @llvm.[[TARGET]].thread.id(i32 0)
  17 // CHECK:       %[[#ID_X_:]] = insertelement <2 x i32> poison, i32 %[[#ID_X]], i64 0
  18 // CHECK:       %[[#ID_Y:]] = call i32 @llvm.[[TARGET]].thread.id(i32 1)
  19 // CHECK:       %[[#ID_XY:]] = insertelement <2 x i32> %[[#ID_X_]], i32 %[[#ID_Y]], i64 1
  20 // CHECK-DXIL:  call void @{{.*}}bar{{.*}}(<2 x i32> %[[#ID_XY]])
  21 // CHECK-SPIRV:  call spir_func void @{{.*}}bar{{.*}}(<2 x i32> %[[#ID_XY]])
  22 [shader("compute")]
  23 [numthreads(8,8,1)]
  24 void bar(uint2 Idx : SV_DispatchThreadID) {}
  25