Sean Keely
1fc7f2dec7
Change-Id: I53c1845d985ac3e9708d952865009c0021f3bb4f
[ROCm/ROCR-Runtime commit: 7e3db20826]
54 lines
2.1 KiB
C++
Executable File
54 lines
2.1 KiB
C++
Executable File
#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_
|