sglang_v0.5.2/pytorch_2.8.0/third_party/fbgemm/src
hailin c8e8c1e9ff . 2025-09-20 16:09:34 +08:00
..
fp32 . 2025-09-20 16:09:34 +08:00
CodeCache.h . 2025-09-20 16:09:34 +08:00
CodeGenHelpers.h . 2025-09-20 16:09:34 +08:00
DirectConv.h . 2025-09-20 16:09:34 +08:00
EmbeddingSpMDM.cc . 2025-09-20 16:09:34 +08:00
EmbeddingSpMDMAutovec.cc . 2025-09-20 16:09:34 +08:00
EmbeddingSpMDMAutovec.h . 2025-09-20 16:09:34 +08:00
EmbeddingSpMDMAvx2.cc . 2025-09-20 16:09:34 +08:00
EmbeddingSpMDMAvx512.cc . 2025-09-20 16:09:34 +08:00
EmbeddingSpMDMNBit.cc . 2025-09-20 16:09:34 +08:00
ExecuteKernel.cc . 2025-09-20 16:09:34 +08:00
ExecuteKernel.h . 2025-09-20 16:09:34 +08:00
ExecuteKernelGeneric.h . 2025-09-20 16:09:34 +08:00
ExecuteKernelU8S8.cc . 2025-09-20 16:09:34 +08:00
ExecuteKernelU8S8.h . 2025-09-20 16:09:34 +08:00
Fbgemm.cc . 2025-09-20 16:09:34 +08:00
FbgemmBfloat16Convert.cc . 2025-09-20 16:09:34 +08:00
FbgemmBfloat16ConvertAvx2.cc . 2025-09-20 16:09:34 +08:00
FbgemmBfloat16ConvertAvx512.cc . 2025-09-20 16:09:34 +08:00
FbgemmConv.cc . 2025-09-20 16:09:34 +08:00
FbgemmFP16.cc . 2025-09-20 16:09:34 +08:00
FbgemmFP16UKernelsAvx2.cc . 2025-09-20 16:09:34 +08:00
FbgemmFP16UKernelsAvx2.h . 2025-09-20 16:09:34 +08:00
FbgemmFP16UKernelsAvx512.cc . 2025-09-20 16:09:34 +08:00
FbgemmFP16UKernelsAvx512.h . 2025-09-20 16:09:34 +08:00
FbgemmFP16UKernelsAvx512_256.cc . 2025-09-20 16:09:34 +08:00
FbgemmFP16UKernelsAvx512_256.h . 2025-09-20 16:09:34 +08:00
FbgemmFP16UKernelsIntrinsicAvx2.cc . 2025-09-20 16:09:34 +08:00
FbgemmFP16UKernelsIntrinsicAvx512.cc . 2025-09-20 16:09:34 +08:00
FbgemmFP16UKernelsIntrinsicAvx512_256.cc . 2025-09-20 16:09:34 +08:00
FbgemmFP16UKernelsSve128.cc . 2025-09-20 16:09:34 +08:00
FbgemmFP16UKernelsSve128.h . 2025-09-20 16:09:34 +08:00
FbgemmFPCommon.cc . 2025-09-20 16:09:34 +08:00
FbgemmFloat16Convert.cc . 2025-09-20 16:09:34 +08:00
FbgemmFloat16ConvertAvx2.cc . 2025-09-20 16:09:34 +08:00
FbgemmFloat16ConvertAvx512.cc . 2025-09-20 16:09:34 +08:00
FbgemmFloat16ConvertSVE.cc . 2025-09-20 16:09:34 +08:00
FbgemmI8Depthwise2DAvx2-inl.h . 2025-09-20 16:09:34 +08:00
FbgemmI8Depthwise3DAvx2.cc . 2025-09-20 16:09:34 +08:00
FbgemmI8DepthwiseAvx2-inl.h . 2025-09-20 16:09:34 +08:00
FbgemmI8DepthwiseAvx2.cc . 2025-09-20 16:09:34 +08:00
FbgemmI8DepthwisePerChannelQuantAvx2.cc . 2025-09-20 16:09:34 +08:00
FbgemmI8Spmdm.cc . 2025-09-20 16:09:34 +08:00
FbgemmI64.cc . 2025-09-20 16:09:34 +08:00
FbgemmPackMatrixB.cc . 2025-09-20 16:09:34 +08:00
FbgemmSparseDense.cc . 2025-09-20 16:09:34 +08:00
FbgemmSparseDenseAvx2.cc . 2025-09-20 16:09:34 +08:00
FbgemmSparseDenseAvx512.cc . 2025-09-20 16:09:34 +08:00
FbgemmSparseDenseInt8Avx2.cc . 2025-09-20 16:09:34 +08:00
FbgemmSparseDenseInt8Avx512.cc . 2025-09-20 16:09:34 +08:00
FbgemmSparseDenseVectorInt8Avx512.cc . 2025-09-20 16:09:34 +08:00
GenerateI8Depthwise.cc . 2025-09-20 16:09:34 +08:00
GenerateI8Depthwise.h . 2025-09-20 16:09:34 +08:00
GenerateKernel.cc . 2025-09-20 16:09:34 +08:00
GenerateKernel.h . 2025-09-20 16:09:34 +08:00
GenerateKernelDirectConvU8S8S32ACC32.cc . 2025-09-20 16:09:34 +08:00
GenerateKernelU8S8S32ACC16.cc . 2025-09-20 16:09:34 +08:00
GenerateKernelU8S8S32ACC16Avx512.cc . 2025-09-20 16:09:34 +08:00
GenerateKernelU8S8S32ACC16Avx512VNNI.cc . 2025-09-20 16:09:34 +08:00
GenerateKernelU8S8S32ACC32.cc . 2025-09-20 16:09:34 +08:00
GenerateKernelU8S8S32ACC32Avx512VNNI.cc . 2025-09-20 16:09:34 +08:00
GroupwiseConv.cc . 2025-09-20 16:09:34 +08:00
GroupwiseConv.h . 2025-09-20 16:09:34 +08:00
GroupwiseConvAcc32Avx2.cc . 2025-09-20 16:09:34 +08:00
GroupwiseConvAcc32Avx512.cc . 2025-09-20 16:09:34 +08:00
InlineAsmDefines.h . 2025-09-20 16:09:34 +08:00
KleidiAIFP16UKernelsNeon.cc . 2025-09-20 16:09:34 +08:00
KleidiAIFP16UKernelsNeon.h . 2025-09-20 16:09:34 +08:00
MaskAvx2.h . 2025-09-20 16:09:34 +08:00
OptimizedKernelsAvx2.cc . 2025-09-20 16:09:34 +08:00
OptimizedKernelsAvx2.h . 2025-09-20 16:09:34 +08:00
PackAMatrix.cc . 2025-09-20 16:09:34 +08:00
PackAWithIm2Col.cc . 2025-09-20 16:09:34 +08:00
PackAWithQuantRowOffset.cc . 2025-09-20 16:09:34 +08:00
PackAWithRowOffset.cc . 2025-09-20 16:09:34 +08:00
PackBMatrix.cc . 2025-09-20 16:09:34 +08:00
PackDepthwiseConvMatrixAvx2.cc . 2025-09-20 16:09:34 +08:00
PackMatrix.cc . 2025-09-20 16:09:34 +08:00
PackWeightMatrixForGConv.cc . 2025-09-20 16:09:34 +08:00
PackWeightsForConv.cc . 2025-09-20 16:09:34 +08:00
PackWeightsForDirectConv.cc . 2025-09-20 16:09:34 +08:00
QuantUtils.cc . 2025-09-20 16:09:34 +08:00
QuantUtilsAvx2.cc . 2025-09-20 16:09:34 +08:00
QuantUtilsAvx512.cc . 2025-09-20 16:09:34 +08:00
QuantUtilsNeon.cc . 2025-09-20 16:09:34 +08:00
RefImplementations.cc . 2025-09-20 16:09:34 +08:00
RefImplementations.h . 2025-09-20 16:09:34 +08:00
RowWiseSparseAdagradFused.cc . 2025-09-20 16:09:34 +08:00
SparseAdagrad.cc . 2025-09-20 16:09:34 +08:00
TransposeUtils.cc . 2025-09-20 16:09:34 +08:00
TransposeUtils.h . 2025-09-20 16:09:34 +08:00
TransposeUtilsAvx2.h . 2025-09-20 16:09:34 +08:00
TransposeUtilsNeon.h . 2025-09-20 16:09:34 +08:00
TransposeUtilsSve.h . 2025-09-20 16:09:34 +08:00
Utils.cc . 2025-09-20 16:09:34 +08:00
UtilsAvx2.cc . 2025-09-20 16:09:34 +08:00
UtilsAvx512.cc . 2025-09-20 16:09:34 +08:00
UtilsNeon.cc . 2025-09-20 16:09:34 +08:00
UtilsSve.cc . 2025-09-20 16:09:34 +08:00
codegen_fp16fp32.cc . 2025-09-20 16:09:34 +08:00
spmmUtils.cc . 2025-09-20 16:09:34 +08:00
spmmUtilsAvx2.cc . 2025-09-20 16:09:34 +08:00