Function check_equal_gpuid

detrsmpl/core/renderer/mpr_renderer/cuda/rasterizer.cpp:36–41

}

void check_equal_gpuid(const torch::Tensor& a, const torch::Tensor& b) {
    TORCH_CHECK(
        a.device().index() == b.device().index(),
        "expected equal gpu id, got ", a.device().index(), " != ", b.device().index()
    );
}

std::vector<torch::Tensor> estimate_normals(
    const torch::Tensor& vertices_ndc,