Laurent Morichetti
99 строки
3.1 KiB
C++
99 строки
3.1 KiB
C++
/* Copyright (c) 2010-present Advanced Micro Devices, Inc.
|
|
|
|
Permission is hereby granted, free of charge, to any person obtaining a copy
|
|
of this software and associated documentation files (the "Software"), to deal
|
|
in the Software without restriction, including without limitation the rights
|
|
to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
|
|
copies of the Software, and to permit persons to whom the Software is
|
|
furnished to do so, subject to the following conditions:
|
|
|
|
The above copyright notice and this permission notice shall be included in
|
|
all copies or substantial portions of the Software.
|
|
|
|
THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
|
|
IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
|
|
FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
|
|
AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
|
|
LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
|
|
OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
|
|
THE SOFTWARE. */
|
|
|
|
#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
|