Disable mid-thread preemption for VC kernels using

hw thread id

.

Author: vsemenov368

IGC/VectorCompiler/igcdeps/src/cmc.cpp

@@ -827,6 +827,7 @@ static void setExecutionInfo(const GenXOCLRuntimeInfo::KernelInfo &BackendInfo,
   ExecEnv.HasBarriers = BackendInfo.getNumBarriers();
   ExecEnv.HasSample = BackendInfo.usesSample();
   ExecEnv.HasDPAS = BackendInfo.usesDPAS();
+  ExecEnv.DisableMidThreadPreemption = BackendInfo.disableMidThreadPreemption();
   ExecEnv.numThreads = BackendInfo.getNumThreads();
   ExecEnv.HasReadWriteImages = BackendInfo.usesReadWriteImages();
   ExecEnv.SubgroupIndependentForwardProgressRequired = true;

IGC/VectorCompiler/include/vc/GenXCodeGen/GenXOCLRuntimeInfo.h

@@ -179,6 +179,7 @@ class GenXOCLRuntimeInfo : public ModulePass {
     bool UsesGroupId = false;
     bool UsesReadWriteImages = false;
     bool UsesSample = false;
+    bool DisableMidThreadPreemption = false;
 
     unsigned GRFSizeInBytes;
     unsigned NumBarriers = 0;
@@ -268,6 +269,10 @@ class GenXOCLRuntimeInfo : public ModulePass {
     // igcmc always sets this to zero. Preserve this here.
     unsigned getNumThreads() const { return 0; }
 
+    bool disableMidThreadPreemption() const {
+      return FuncInfo.DisableMidThreadPreemption;
+    }
+
     unsigned getNumBarriers() const { return FuncInfo.NumBarriers; }
     bool usesSample() const { return FuncInfo.UsesSample; }
     bool usesReadWriteImages() const { return FuncInfo.UsesReadWriteImages; }

IGC/VectorCompiler/include/vc/Utils/GenX/InternalMetadata.h

@@ -1,6 +1,6 @@
 /*========================== begin_copyright_notice ============================
 
-Copyright (C) 2021-2023 Intel Corporation
+Copyright (C) 2021-2024 Intel Corporation
 
 SPDX-License-Identifier: MIT
 
@@ -18,6 +18,8 @@ namespace vc {
 namespace FunctionMD {
 inline constexpr const char GenXKernelInternal[] = "genx.kernel.internal";
 inline constexpr const char VCBuiltinFunction[] = "VC.Builtin";
+inline constexpr const char VCDisableMidThreadPreemption[] =
+    "vc.disable.mid.thread.preemption";
 
 // amount of stack calculated for kernel
 // no attribute means that GenXStackUsage pass failed (recursion, etc)

IGC/VectorCompiler/lib/GenXCodeGen/GenXLowering.cpp

@@ -122,6 +122,7 @@ SPDX-License-Identifier: MIT
 
 #include "vc/Support/GenXDiagnostic.h"
 #include "vc/Utils/GenX/GlobalVariable.h"
+#include "vc/Utils/GenX/InternalMetadata.h"
 
 #include "Probe/Assertion.h"
 #include <algorithm>
@@ -4798,6 +4799,9 @@ bool GenXLowering::lowerHardwareThreadID(CallInst *CI) {
     Res = IRB.CreateAnd(Res, MaskC);
   }
 
+  CI->getModule()->getOrInsertNamedMetadata(
+      vc::FunctionMD::VCDisableMidThreadPreemption);
+
   CI->replaceAllUsesWith(Res);
   ToErase.push_back(CI);
   return true;

IGC/VectorCompiler/lib/GenXCodeGen/GenXOCLRuntimeInfo.cpp

@@ -324,6 +324,11 @@ GenXOCLRuntimeInfo::FunctionInfo::FunctionInfo(const FunctionGroup &FG,
     SLMSize = KernelMD.getSLMSize();
     IndirectCount = KernelMD.getIndirectCount();
 
+    if (Func->getParent()->getNamedMetadata(
+            FunctionMD::VCDisableMidThreadPreemption)) {
+      DisableMidThreadPreemption = true;
+    }
+
     if (ST.hasNBarrier())
       NumBarriers = KernelMD.getAlignedBarrierCnt(NumBarriers);
   } else {

IGC/VectorCompiler/test/Lowering/get_hwid.ll

@@ -1,6 +1,6 @@
 ;=========================== begin_copyright_notice ============================
 ;
-; Copyright (C) 2023 Intel Corporation
+; Copyright (C) 2023-2024 Intel Corporation
 ;
 ; SPDX-License-Identifier: MIT
 ;
@@ -14,28 +14,28 @@
 ; COM:  - Xe2-like concatination (CHECK-Xe2): Xe2
 
 ; RUN: %opt %use_old_pass_manager% -GenXLowering -march=genx64 -mcpu=Gen9 -mtriple=spir64-unknown-unknown -S < %s | \
-; RUN: FileCheck %s --check-prefix=CHECK-PREDEF
+; RUN: FileCheck %s --check-prefixes=CHECK-PREDEF,CHECK
 
 ; RUN: %opt %use_old_pass_manager% -GenXLowering -march=genx64 -mcpu=Gen11 -mtriple=spir64-unknown-unknown -S < %s | \
-; RUN: FileCheck %s --check-prefix=CHECK-PREDEF
+; RUN: FileCheck %s --check-prefixes=CHECK-PREDEF,CHECK
 
 ; RUN: %opt %use_old_pass_manager% -GenXLowering -march=genx64 -mcpu=XeLP -mtriple=spir64-unknown-unknown -S < %s | \
-; RUN: FileCheck %s --check-prefix=CHECK-PREDEF
+; RUN: FileCheck %s --check-prefixes=CHECK-PREDEF,CHECK
 
 ; RUN: %opt %use_old_pass_manager% -GenXLowering -march=genx64 -mcpu=XeHP -mtriple=spir64-unknown-unknown -S < %s | \
-; RUN: FileCheck %s --check-prefix=CHECK-XeHP
+; RUN: FileCheck %s --check-prefixes=CHECK-XeHP,CHECK
 ; RUN: %opt %use_old_pass_manager% -GenXLowering -march=genx64 -mcpu=XeHPG -mtriple=spir64-unknown-unknown -S < %s | \
-; RUN: FileCheck %s --check-prefix=CHECK-XeHP
+; RUN: FileCheck %s --check-prefixes=CHECK-XeHP,CHECK
 ; RUN: %opt %use_old_pass_manager% -GenXLowering -march=genx64 -mcpu=XeLPG -mtriple=spir64-unknown-unknown -S < %s | \
-; RUN: FileCheck %s --check-prefix=CHECK-XeHP
+; RUN: FileCheck %s --check-prefixes=CHECK-XeHP,CHECK
 ; RUN: %opt %use_old_pass_manager% -GenXLowering -march=genx64 -mcpu=XeLPGPlus -mtriple=spir64-unknown-unknown -S < %s | \
-; RUN: FileCheck %s --check-prefix=CHECK-XeHP
+; RUN: FileCheck %s --check-prefixes=CHECK-XeHP,CHECK
 
 ; RUN: %opt %use_old_pass_manager% -GenXLowering -march=genx64 -mcpu=XeHPC -mtriple=spir64-unknown-unknown -S < %s | \
-; RUN: FileCheck %s --check-prefix=CHECK-XeHPC
+; RUN: FileCheck %s --check-prefixes=CHECK-XeHPC,CHECK
 
 ; RUN: %opt %use_old_pass_manager% -GenXLowering -march=genx64 -mcpu=Xe2 -mtriple=spir64-unknown-unknown -S < %s | \
-; RUN: FileCheck %s --check-prefix=CHECK-Xe2
+; RUN: FileCheck %s --check-prefixes=CHECK-Xe2,CHECK
 
 ; CHECK-PREDEF: [[RES:%[^ ]+]] = call i32 @llvm.genx.read.predef.reg.i32.i32(i32 12, i32 undef)
 ; CHECK-PREDEF: ret i32 [[RES]]
@@ -80,6 +80,8 @@
 ; CHECK-Xe2: [[RES:%[^ ]+]] = and i32 [[OR2]], 8191
 ; CHECK-Xe2: ret i32 [[RES]]
 
+; CHECK: !vc.disable.mid.thread.preemption = !{}
+
 target datalayout = "e-p:64:64-i64:64-n8:16:32"
 target triple = "genx64-unknown-unknown"