Move Images code to hsa-runtime folder

Change-Id: I53c1845d985ac3e9708d952865009c0021f3bb4f [ROCm/ROCR-Runtime commit: 7e3db20826]
2020-04-30 00:18:36 -05:00
parent bd5ef0eff8
commit 1fc7f2dec7
70 changed files with 73713 additions and 47 deletions
@@ -0,0 +1,53 @@
+#ifndef EXT_IMAGE_IMAGE_MANAGER_NV_H_ 
+#define EXT_IMAGE_IMAGE_MANAGER_NV_H_ 
+
+#include "addrlib/inc/addrinterface.h"
+#include "image_manager_kv.h"
+
+namespace amd {
+
+class ImageManagerNv : public ImageManagerKv {
+ public:
+  ImageManagerNv();
+  virtual ~ImageManagerNv();
+
+  /// @brief Calculate the size and alignment of the backing storage of an
+  /// image.
+  virtual hsa_status_t CalculateImageSizeAndAlignment(
+      hsa_agent_t component, const hsa_ext_image_descriptor_t& desc,
+      hsa_ext_image_data_layout_t image_data_layout,
+      size_t image_data_row_pitch, size_t image_data_slice_pitch,
+      hsa_ext_image_data_info_t& image_info) const;
+
+  /// @brief Fill image structure with device specific image object.
+  virtual hsa_status_t PopulateImageSrd(Image& image) const;
+
+  /// @brief Fill image structure with device specific image object using the given format.
+  virtual hsa_status_t PopulateImageSrd(Image& image, const metadata_amd_t* desc) const;
+
+  /// @brief Modify device specific image object according to the specified
+  /// new format.
+  virtual hsa_status_t ModifyImageSrd(Image& image,
+                                      hsa_ext_image_format_t& new_format) const;
+
+  /// @brief Fill sampler structure with device specific sampler object.
+  virtual hsa_status_t PopulateSamplerSrd(Sampler& sampler) const;
+
+  /// @brief Fill image backing storage using agent copy.
+  virtual hsa_status_t FillImage(const Image& image, const void* pattern,
+                                 const hsa_ext_image_region_t& region);
+ protected:
+  uint32_t GetAddrlibSurfaceInfoNv(hsa_agent_t component,
+                             const hsa_ext_image_descriptor_t& desc,
+                             Image::TileMode tileMode,
+                             size_t image_data_row_pitch,
+                             size_t image_data_slice_pitch,
+                             ADDR2_COMPUTE_SURFACE_INFO_OUTPUT& out) const;
+
+  bool IsLocalMemory(const void* address) const;
+
+ private:
+  DISALLOW_COPY_AND_ASSIGN(ImageManagerNv);
+};  // namespace amd
+}
+#endif  // EXT_IMAGE_IMAGE_MANAGER_NV_H_