rocm-systems/rocclr/runtime/device/gpu/gpuconstbuf.cpp

//
// Copyright (c) 2010 Advanced Micro Devices, Inc. All rights reserved.
//

#include "device/gpu/gpuconstbuf.hpp"
#include "device/gpu/gpuvirtual.hpp"
#include "device/gpu/gpudevice.hpp"
#include "device/gpu/gpusettings.hpp"

namespace gpu {

ConstBuffer::ConstBuffer(VirtualGPU& gpu, size_t size)
    : Memory(const_cast<gpu::Device&>(gpu.dev()), size * VectorSize),
      gpu_(gpu),
      size_(size * VectorSize),
      wrtOffset_(0),
      lastWrtSize_(0),
      wrtAddress_(NULL) {}

ConstBuffer::~ConstBuffer() {
  if (wrtAddress_ != NULL) {
    unmap(&gpu_);
  }

  amd::AlignedMemory::deallocate(sysMemCopy_);
}

bool ConstBuffer::create() {
  // Create sysmem copy for the constant buffer
  sysMemCopy_ = reinterpret_cast<address>(amd::AlignedMemory::allocate(size_, 256));
  if (sysMemCopy_ == NULL) {
    LogPrintfError(
        "We couldn't allocate sysmem copy for constant buffer,\
            size(%d)!",
        size_);
    return false;
  }
  memset(sysMemCopy_, 0, size_);

  if (!Memory::create(Resource::RemoteUSWC)) {
    LogPrintfError("We couldn't create HW constant buffer, size(%d)!", size_);
    return false;
  }

  // Constant buffer warm-up
  warmUpRenames(gpu_);

  wrtAddress_ = map(&gpu_, Resource::Discard);
  if (wrtAddress_ == NULL) {
    LogPrintfError("We couldn't map HW constant buffer, size(%d)!", size_);
    return false;
  }

  return true;
}

bool ConstBuffer::uploadDataToHw(size_t size) {
  static const size_t HwCbAlignment = 256;

  // Align copy size on the vector's boundary
  size_t count = amd::alignUp(size, VectorSize);
  wrtOffset_ += lastWrtSize_;

  // Check if CB has enough space for copy
  if ((wrtOffset_ + count) > size_) {
    if (wrtAddress_ != NULL) {
      unmap(&gpu_);
    }
    wrtAddress_ = map(&gpu_, Resource::Discard);
    wrtOffset_ = 0;
    lastWrtSize_ = 0;
  }

  // Update memory with new CB data
  memcpy((reinterpret_cast<char*>(wrtAddress_) + wrtOffset_), sysMemCopy_, count);

  // Adjust the size by the HW CB buffer alignment
  lastWrtSize_ = amd::alignUp(size, HwCbAlignment);
  return true;
}

}  // namespace gpu