Allocate device binary images in-place, not in cache

Signed-off-by: Sergey Kanaev <sergey.kanaev@intel.com>

Author: Sergey Kanaev

sycl/unittests/helpers/PiImage.hpp

@@ -233,46 +233,30 @@ class PiImage {
 
 /// Convenience wrapper around pi_device_binaries_struct, that manages mock
 /// device images' lifecycle.
+template <size_t __NumberOfImages>
 class PiImageArray {
 public:
-  /// Constructs an array of device images from a single image and registers
-  /// it with SYCL runtime.
-  PiImageArray(PiImage Image) {
-    MImages.push_back(std::move(Image));
-    convertImages();
-    MAllBinaries = pi_device_binaries_struct{
-        PI_DEVICE_BINARIES_VERSION,
-        1, // num binaries
-        MNativeImages.data(),
-        nullptr, // not used, for compatibility with OpenMP
-        nullptr  // not used, for compatibility with OpenMP
-    };
-    __sycl_register_lib(&MAllBinaries);
-  }
+  static constexpr size_t NumberOfImages = __NumberOfImages;
+
+  PiImageArray(const PiImage *Imgs) {
+    for (size_t Idx = 0; Idx < NumberOfImages; ++Idx)
+      MNativeImages[Idx] = Imgs[Idx].convertToNativeType();
 
-  /// Constructs an array of device images and registers it with SYCL runtime.
-  PiImageArray(std::vector<PiImage> Images) : MImages(std::move(Images)) {
-    convertImages();
     MAllBinaries = pi_device_binaries_struct{
         PI_DEVICE_BINARIES_VERSION,
-        static_cast<uint16_t>(MNativeImages.size()), // num binaries
-        MNativeImages.data(),
-        nullptr, // not used, for compatibility with OpenMP
-        nullptr  // not used, for compatibility with OpenMP
+        NumberOfImages,
+        MNativeImages,
+        nullptr, // not used, put here for compatibility with OpenMP
+        nullptr, // not used, put here for compatibility with OpenMP
     };
+
     __sycl_register_lib(&MAllBinaries);
   }
 
   ~PiImageArray() { __sycl_unregister_lib(&MAllBinaries); }
 
 private:
-  void convertImages() {
-    std::transform(
-        MImages.begin(), MImages.end(), std::back_inserter(MNativeImages),
-        [](const PiImage &Img) { return Img.convertToNativeType(); });
-  }
-  std::vector<PiImage> MImages;
-  std::vector<pi_device_binary_struct> MNativeImages;
+  pi_device_binary_struct MNativeImages[NumberOfImages];
   pi_device_binaries_struct MAllBinaries;
 };
 

sycl/unittests/misc/KernelBuildOptions.cpp

@@ -208,7 +208,7 @@ static sycl::unittest::PiImage generateDefaultImage() {
 }
 
 sycl::unittest::PiImage Img = generateDefaultImage();
-sycl::unittest::PiImageArray ImgArray{Img};
+sycl::unittest::PiImageArray<1> ImgArray{&Img};
 
 TEST(KernelBuildOptions, KernelBundleBasic) {
   sycl::platform Plt{sycl::default_selector()};

sycl/unittests/program_manager/itt_annotations.cpp

@@ -239,7 +239,7 @@ static sycl::unittest::PiImage generateDefaultImage() {
 }
 
 sycl::unittest::PiImage Img = generateDefaultImage();
-sycl::unittest::PiImageArray ImgArray{Img};
+sycl::unittest::PiImageArray<1> ImgArray{&Img};
 
 TEST(ITTNotify, UseKernelBundle) {
   set_env(ITTProfileEnvVarName, "1");

sycl/unittests/scheduler/RequiredWGSize.cpp

@@ -226,7 +226,7 @@ static sycl::unittest::PiImage generateDefaultImage() {
 }
 
 sycl::unittest::PiImage Img = generateDefaultImage();
-sycl::unittest::PiImageArray ImgArray{Img};
+sycl::unittest::PiImageArray<1> ImgArray{&Img};
 
 static void performChecks() {
   sycl::platform Plt{sycl::default_selector()};

sycl/unittests/spec_constants/DefaultValues.cpp

@@ -203,7 +203,7 @@ static sycl::unittest::PiImage generateDefaultImage() {
 }
 
 sycl::unittest::PiImage Img = generateDefaultImage();
-sycl::unittest::PiImageArray ImgArray{Img};
+sycl::unittest::PiImageArray<1> ImgArray{&Img};
 
 TEST(DefaultValues, DISABLED_DefaultValuesAreSet) {
   sycl::platform Plt{sycl::default_selector()};