Mirror/waifu2x-caffe
1
0
Fork 0
mirror of https://github.com/lltcggie/waifu2x-caffe.git synced 2025-06-26 21:52:49 +00:00
Code Issues Packages Projects Releases Wiki Activity

CPUモードで実行すると強制終了するバグを修正

Browse Source
This commit is contained in:
lltcggie 2016-07-06 12:27:03 +09:00
parent 597e5b3164
commit c4a8e4d6db
1 changed files with 6 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
common/waifu2x.cpp
View File

@ -732,11 +732,15 @@ Waifu2x::eWaifu2xError Waifu2x::ProcessNet(std::shared_ptr<cNet> net, const int
if (OutputMemorySize > mOutputBlockSize) if (OutputMemorySize > mOutputBlockSize)
{ {
if (mIsCuda) if (mIsCuda)
{
CUDA_HOST_SAFE_FREE(mOutputBlock); CUDA_HOST_SAFE_FREE(mOutputBlock);
CUDA_CHECK_WAIFU2X(cudaHostAlloc(&mOutputBlock, OutputMemorySize, cudaHostAllocDefault));
}
else else
{
SAFE_DELETE_WAIFU2X(mOutputBlock); SAFE_DELETE_WAIFU2X(mOutputBlock);
mOutputBlock = new float[OutputMemorySize];
CUDA_CHECK_WAIFU2X(cudaHostAlloc(&mOutputBlock, OutputMemorySize, cudaHostAllocDefault)); }
mOutputBlockSize = OutputMemorySize; mOutputBlockSize = OutputMemorySize;
} }