rocm-systems/rocclr/runtime/device/pal/palconstbuf.cpp

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

#include "device/pal/palconstbuf.hpp"
#include "device/pal/palvirtual.hpp"
#include "device/pal/paldevice.hpp"
#include "device/pal/palsettings.hpp"

namespace pal {

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

ConstBuffer::~ConstBuffer()
{
    if (wrtAddress_ != nullptr) {
        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_ == nullptr) {
        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_ == nullptr) {
        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_ != nullptr) {
            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 pal