[CPU] remove RECORD_FUNCTION (#23528)
This commit is contained in:
@@ -441,10 +441,6 @@ at::Tensor flash_attn_varlen_func(
|
||||
int64_t max_seqlen_q,
|
||||
int64_t max_seqlen_k,
|
||||
bool causal) {
|
||||
RECORD_FUNCTION(
|
||||
"sgl_kernel::flash_attn_varlen_func",
|
||||
std::vector<c10::IValue>({q, k, v, cu_seqlens_q, cu_seqlens_k, max_seqlen_q, max_seqlen_k, causal}));
|
||||
|
||||
CHECK_LAST_DIM_CONTIGUOUS_INPUT(q);
|
||||
CHECK_LAST_DIM_CONTIGUOUS_INPUT(k);
|
||||
CHECK_LAST_DIM_CONTIGUOUS_INPUT(v);
|
||||
|
||||
Reference in New Issue
Block a user