MCPcopy Create free account

hub / github.com/bytedance/ByteTransformer / functions

Functions317 in github.com/bytedance/ByteTransformer

⨍ Functions 317◇ Types & classes 134
Functionthread_start_row
Need to get the thread start row from the tile iterator
cutlass_contrib/include/cutlass/contrib/epilogue/threadblock/predicated_tile_iterator.h:374
Methodtile_index
Gets the global tile index
cutlass_contrib/include/cutlass/contrib/gemm/kernel/gemm_grouped.h:199
Methodupdate
Lightweight update given a subset of arguments
cutlass_contrib/include/cutlass/contrib/gemm/device/gemm_grouped.h:180
Methodupdate
cutlass_contrib/include/cutlass/contrib/gemm/kernel/gemm_grouped.h:385
Functionvalid
Returns whether access is valid or not
cutlass_contrib/include/cutlass/contrib/transform/threadblock/predicated_tile_access_iterator.h:564
Methodvalid
Returns whether access is valid or not
cutlass_contrib/include/cutlass/contrib/transform/threadblock/predicated_tile_access_iterator.h:778
Methodvalid
Returns whether access is valid or not
cutlass_contrib/include/cutlass/contrib/transform/threadblock/predicated_tile_access_iterator.h:967
Functionvariety_attention_fused_infer
bytetransformer/include/variety_attention_fused.h:803
Functionvariety_wmma_attention_kernel
bytetransformer/include/variety_attention_fused.h:31
Functionvariety_wmma_attention_long_kernel
bytetransformer/include/variety_attention_fused.h:295
Functionvariety_wmma_attention_long_rm_kernel
bytetransformer/include/variety_attention_fused.h:444
Functionvariety_wmma_attention_rm_kernel
bytetransformer/include/variety_attention_fused.h:164
Method~Attention
bytetransformer/include/attention.h:147
Method~BertTransformer
bytetransformer/include/bert_transformer.h:221
Method~BertTransformer
bytetransformer/th_op/ths_op.h:76
Method~CutlassAttention
bytetransformer/include/cutlass_attention.h:117
Method~IBTEncoder
bytetransformer/th_op/bert_transformer_ext.h:32
← previous301–317 of 317, ranked by callers