[ESIMD] Enable variable slm_init support. (#6030)

* [ESIMD] Enable variable slm_init support.

Author: kychendev

llvm/lib/SYCLLowerIR/ESIMD/LowerESIMD.cpp

@@ -637,6 +637,7 @@ class ESIMDIntrinDescTable {
         {"lane_id", {"lane.id", {}}},
         {"test_src_tmpl_arg",
          {"test.src.tmpl.arg", {t(0), t1(1), t8(2), t16(3), t32(4), c8(17)}}},
+        {"slm_init", {"slm.init", {a(0)}}},
     };
   }
 
@@ -1684,12 +1685,14 @@ size_t SYCLLowerESIMDPass::runOnFunction(Function &F,
       // process ESIMD builtins that go through special handling instead of
       // the translation procedure
       // TODO FIXME slm_init should be made top-level __esimd_slm_init
-      if (Name.startswith("__esimd_slm_init")) {
+      if (Name.startswith("__esimd_slm_init") &&
+          isa<ConstantInt>(CI->getArgOperand(0))) {
         // tag the kernel with meta-data SLMSize, and remove this builtin
         translateSLMInit(*CI);
         ToErase.push_back(CI);
         continue;
       }
+
       if (Name.startswith("__esimd_nbarrier_init")) {
         translateNbarrierInit(*CI);
         ToErase.push_back(CI);

llvm/test/SYCLLowerIR/ESIMD/lower_intrins.ll

@@ -282,6 +282,13 @@ define dso_local spir_func <32 x half>  @FUNC_52() {
   ret <32 x half> %d
 }
 
+define dso_local spir_func void  @FUNC_53(i32 %0) {
+; COM: FIXME: to enable this test after VC-intrinsic sync
+; COM: call spir_func void @_Z16__esimd_slm_initj(i32 %0)
+; COM: CHECK: call void @llvm.genx.slm.init(i32 %0)
+  ret void
+}
+
 declare dso_local i32 @_Z15__esimd_lane_idv()
 declare dso_local spir_func <16 x i16> @_Z12__esimd_sminIsLi16EEN2cm3gen13__vector_typeIT_XT0_EE4typeES5_S5_(<16 x i16> %0, <16 x i16> %1)
 declare dso_local spir_func <8 x float> @_Z16__esimd_rdregionIfLi16ELi8ELi0ELi8ELi1ELi0EEN2cm3gen13__vector_typeIT_XT1_EE4typeENS2_IS3_XT0_EE4typeEt(<16 x float> %0, i16 zeroext %1)
@@ -320,6 +327,7 @@ declare dso_local spir_func <16 x float> @_Z12__esimd_rndzILi16EEN2cl4sycl3ext5i
 declare dso_local spir_func <16 x float> @_Z12__esimd_rndeILi16EEN2cl4sycl3ext5intel12experimental5esimd6detail11vector_typeIfXT_EE4typeES9_(<16 x float>)
 declare dso_local spir_func void @_Z25__esimd_test_src_tmpl_argILi3ELi5ELi7ELi11ELi13EEvv()
 declare dso_local spir_func <32 x half> @_Z16__esimd_wrregionIDF16_Li32ELi32ELi0ELi32ELi1ELi32EEN2cl4sycl3ext5intel12experimental5esimd6detail11vector_typeIT_XT0_EE4typeESA_NS7_IS8_XT1_EE4typeEtNS7_ItXT1_EE4typeE(<32 x half>, <32 x half>, i16 zeroext, <32 x i16>)
+declare dso_local spir_func void @_Z16__esimd_slm_initj(i32 %0)
 
 attributes #0 = { "genx_byte_offset"="192" "genx_volatile" }
 

sycl/include/sycl/ext/intel/esimd/detail/memory_intrin.hpp

@@ -530,7 +530,7 @@ __esimd_svm_atomic2(__ESIMD_DNS::vector_type_t<uint64_t, N> addrs,
 }
 #endif // __SYCL_DEVICE_ONLY__
 
-__ESIMD_INTRIN void __esimd_slm_init(size_t size)
+__ESIMD_INTRIN void __esimd_slm_init(uint32_t size)
 #ifdef __SYCL_DEVICE_ONLY__
     ;
 #else