From 8a73d334b8efca8e1b12e2d486e5b4dd48dccc5f Mon Sep 17 00:00:00 2001 From: "PoYen, Chen" Date: Wed, 24 Jul 2024 05:19:05 +0000 Subject: [PATCH] Rename utility function --- example/ck_tile/01_fmha/rotary.hpp | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/example/ck_tile/01_fmha/rotary.hpp b/example/ck_tile/01_fmha/rotary.hpp index 76a282d76c..d8ceb99ecc 100644 --- a/example/ck_tile/01_fmha/rotary.hpp +++ b/example/ck_tile/01_fmha/rotary.hpp @@ -61,10 +61,10 @@ generate_rotary_cos_sin(ck_tile::index_t seqlen_k, template std::tuple, ck_tile::HostTensor> -index_cos_sin(const ck_tile::HostTensor& cos, - const ck_tile::HostTensor& sin, - ck_tile::index_t seqlen_offset, - ck_tile::index_t seqlen) +slice_rotary_cos_sin(const ck_tile::HostTensor& cos, + const ck_tile::HostTensor& sin, + ck_tile::index_t seqlen_offset, + ck_tile::index_t seqlen) { assert(cos.get_num_of_dimension() == 2 && sin.get_num_of_dimension() == 2); assert(cos.get_length(0) == sin.get_length(0) && cos.get_length(1) == sin.get_length(1));