GpuBackend.cpp 文件参考

#include "tensor/GpuBackend.h"
#include "tensor/Graph.h"
#include "tensor/KernelGen.h"
#include "tensor/Optimizer.h"
#include "gpgpu/ComputeShader.h"
#include "gpgpu/Gpgpu.h"
#include "gpgpu/GpuBuffer.h"
#include "gpgpu/Sequence.h"
#include "common/Exception.h"
#include <algorithm>
#include <chrono>
#include <cstdio>
#include <cstring>
#include <limits>
#include <map>
#include <memory>
#include <vector>

浏览源代码.

类
struct	eve::tensor::GpuProgram::Impl
struct	eve::tensor::GpuProgram::Impl::GroupRuntime

命名空间
namespace	eve
namespace	eve::tensor

函数
bool	eve::tensor::gpuReduce (const float *data, int size, int op, float &outResult)
	GPU-accelerated reduction for large eager tensors. op: 0 = sum, 1 = min, 2 = max. Returns false (caller should fall back to CPU) when Vulkan/gpgpu isn't available.

变量说明

gpgpu

gpgpu::Gpgpu* gpgpu = nullptr

在文件 GpuBackend.cpp 第 55 行定义.

被这些函数引用 eve::particles::stepEmitterSimGpu().

reduce

gpgpu::ComputeShader* reduce = nullptr

在文件 GpuBackend.cpp 第 56 行定义.

被这些函数引用 eve::tensor::kernels::argmax(), eve::tensor::kernels::reduceAxis() , 以及 eve::tensor::kernels::softmax().