#pragma once
#include <torch/extension.h>

at::Tensor multires_hash_encode(const int levels, const int coarse_levels, at::Tensor x,
                                at::Tensor res_list, at::Tensor hash_table,
                                at::Tensor hash_table_offsets);

std::tuple<at::Tensor, at::Tensor, at::Tensor>
multires_hash_encode_with_grad(const int levels, const int coarse_levels, at::Tensor x,
                               at::Tensor res_list, at::Tensor hash_table,
                               at::Tensor hash_table_offsets);