Spaces:
Running
on
T4
Running
on
T4
File size: 1,644 Bytes
83d5461 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 |
/*
* Copyright (C) 2023, Inria
* GRAPHDECO research group, https://team.inria.fr/graphdeco
* All rights reserved.
*
* This software is free for non-commercial, research and evaluation use
* under the terms of the LICENSE.md file.
*
* For inquiries contact george.drettakis@inria.fr
*/
#ifndef CUDA_RASTERIZER_BACKWARD_H_INCLUDED
#define CUDA_RASTERIZER_BACKWARD_H_INCLUDED
#include "cuda_runtime.h"
#include "device_launch_parameters.h"
#include <cuda.h>
#define GLM_FORCE_CUDA
#include <glm/glm.hpp>
namespace BACKWARD {
void render(const dim3 grid, dim3 block, const uint2 *ranges,
const uint32_t *point_list, int W, int H, const float *bg_color,
const float2 *means2D, const float4 *conic_opacity,
const float *colors, const float *final_Ts,
const uint32_t *n_contrib, const float *dL_dpixels,
float3 *dL_dmean2D, float4 *dL_dconic2D, float *dL_dopacity,
float *dL_dcolors);
void preprocess(int P, int D, int M, const float3 *means, const int *radii,
const float *shs, const bool *clamped, const glm::vec3 *scales,
const glm::vec4 *rotations, const float scale_modifier,
const float *cov3Ds, const float *view, const float *proj,
const float focal_x, float focal_y, const float tan_fovx,
float tan_fovy, const glm::vec3 *campos,
const float3 *dL_dmean2D, const float *dL_dconics,
glm::vec3 *dL_dmeans, float *dL_dcolor, float *dL_dcov3D,
float *dL_dsh, glm::vec3 *dL_dscale, glm::vec4 *dL_drot);
} // namespace BACKWARD
#endif |