Unverified Commit dac6a3f6 authored by Steve Grubb's avatar Steve Grubb Committed by GitHub
Browse files

[Misc] Apply a couple g++ cleanups (#4719)

parent 64b77dfd
......@@ -84,7 +84,7 @@ void reshape_and_cache_cpu_impl(
void copy_blocks(std::vector<torch::Tensor> &key_caches,
std::vector<torch::Tensor> &value_caches,
const torch::Tensor& block_mapping) {
int num_layers = key_caches.size();
unsigned num_layers = key_caches.size();
TORCH_CHECK(num_layers == value_caches.size());
if (num_layers == 0) {
return;
......
......@@ -19,7 +19,6 @@ void rotary_embedding_impl(
const int num_tokens) {
using scalar_vec_t = vec_op::vec_t<scalar_t>;
constexpr int VEC_ELEM_NUM = scalar_vec_t::get_elem_num();
constexpr int ELEM_SIZE = sizeof(scalar_t);
const int embed_dim = rot_dim / 2;
TORCH_CHECK(embed_dim % VEC_ELEM_NUM == 0);
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment