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

This commit is contained in:
Steve Grubb
2024-05-10 09:37:05 -04:00
committed by GitHub
parent 64b77dfd7e
commit dac6a3f6ed
2 changed files with 1 additions and 2 deletions

View File

@ -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);