583f2bdd13
BOMs barf on UTF-8 decoding within PyTorch:
File "D:\Soft\Miniconda3\lib\site-packages\torch\utils_cpp_extension_versioner.py", line 16, in hash_source_files
hash_value = update_hash(hash_value, file.read())
UnicodeDecodeError: 'gbk' codec can't decode byte 0xbf in position 2: illegal multibyte sequence
Should fix #10, #14
59 lines
1.8 KiB
C++
59 lines
1.8 KiB
C++
// Copyright (c) 2021, NVIDIA CORPORATION & AFFILIATES. All rights reserved.
|
|
//
|
|
// NVIDIA CORPORATION and its licensors retain all intellectual property
|
|
// and proprietary rights in and to this software, related documentation
|
|
// and any modifications thereto. Any use, reproduction, disclosure or
|
|
// distribution of this software and related documentation without an express
|
|
// license agreement from NVIDIA CORPORATION is strictly prohibited.
|
|
|
|
#include <cuda_runtime.h>
|
|
|
|
//------------------------------------------------------------------------
|
|
// CUDA kernel parameters.
|
|
|
|
struct upfirdn2d_kernel_params
|
|
{
|
|
const void* x;
|
|
const float* f;
|
|
void* y;
|
|
|
|
int2 up;
|
|
int2 down;
|
|
int2 pad0;
|
|
int flip;
|
|
float gain;
|
|
|
|
int4 inSize; // [width, height, channel, batch]
|
|
int4 inStride;
|
|
int2 filterSize; // [width, height]
|
|
int2 filterStride;
|
|
int4 outSize; // [width, height, channel, batch]
|
|
int4 outStride;
|
|
int sizeMinor;
|
|
int sizeMajor;
|
|
|
|
int loopMinor;
|
|
int loopMajor;
|
|
int loopX;
|
|
int launchMinor;
|
|
int launchMajor;
|
|
};
|
|
|
|
//------------------------------------------------------------------------
|
|
// CUDA kernel specialization.
|
|
|
|
struct upfirdn2d_kernel_spec
|
|
{
|
|
void* kernel;
|
|
int tileOutW;
|
|
int tileOutH;
|
|
int loopMinor;
|
|
int loopX;
|
|
};
|
|
|
|
//------------------------------------------------------------------------
|
|
// CUDA kernel selection.
|
|
|
|
template <class T> upfirdn2d_kernel_spec choose_upfirdn2d_kernel(const upfirdn2d_kernel_params& p);
|
|
|
|
//------------------------------------------------------------------------
|