Added new API to hide the validation check on amd specific attribute

Author: Kewen12

offload/plugins-nextgen/amdgpu/src/rtl.cpp

@@ -853,6 +853,10 @@ struct AMDGPUKernelTy : public GenericKernelTy {
   /// Indicates whether or not we need to set up our own private segment size.
   bool usesDynamicStack() const { return DynamicStack; }
 
+  bool isValidBlockSize(uint32_t BlockSize) const override {
+    return BlockSize <= ConstWGSize;
+  }
+
   /// Envar to enable occupancy-based optimization for SPMD kernel.
   BoolEnvar OMPX_SPMDOccupancyBasedOpt;
 

offload/plugins-nextgen/common/include/PluginInterface.h

@@ -367,6 +367,9 @@ struct GenericKernelTy {
     return ExecutionMode == OMP_TGT_EXEC_MODE_XTEAM_RED;
   }
 
+  /// Indicate if the input block size is within the limit.
+  virtual bool isValidBlockSize(uint32_t BlockSize) const { return true; }
+
 protected:
   /// Get the execution mode name of the kernel.
   const char *getExecutionModeName() const {
@@ -1345,8 +1348,10 @@ struct KernelRunRecordTy {
 
   // Get parameters for next kernel launch.
   std::pair<uint32_t, uint32_t>
-  getLaunchParamsForKernel(std::string KernelName,
+  getLaunchParamsForKernel(const GenericKernelTy &Kernel,
                            GenericDeviceTy &GenericDevice) {
+    std::string KernelName = Kernel.getName();
+
     // If the kernel reaches the run limit,
     // return the current optimal launch parameters.
     if (reachedRunLimitForKernel(KernelName)) {
@@ -1381,7 +1386,7 @@ struct KernelRunRecordTy {
 
     // Threads should be smaller than ConstWGSize.
     if (IdxThread >= ThreadCandidate.size() ||
-        ThreadCandidate[IdxThread] >= ConstWGSize) {
+        !Kernel.isValidBlockSize(ThreadCandidate[IdxThread])) {
       TuningData[KernelName].IdxThread = 0;
       TuningData[KernelName].IdxCUMultiplier++;
     }

offload/plugins-nextgen/common/src/PluginInterface.cpp

@@ -740,7 +740,7 @@ Error GenericKernelTy::launch(GenericDeviceTy &GenericDevice, void **ArgPtrs,
            "Autotuning is enabled, but KernelRunRecord is not initialized!");
 
     auto [Teams, Threads] =
-        KernelRecord->getLaunchParamsForKernel(KernelName, GenericDevice);
+        KernelRecord->getLaunchParamsForKernel(*this, GenericDevice);
     NumBlocks[0] = Teams;
     NumThreads[0] = Threads;
   } else {