pytorch
4487 строк · 181.0 Кб
1//// --------------------------2//// ATTENTION:3//// THIS CODE IS AUTOGENERATED4//// BY hp_emblookup_codegen.py5//// DO NOT MODIFY!!!6//// --------------------------7}411bool EmbeddingLookupIdx_int32_t_float_float_false__avx2_fma(412const int64_t block_size,413const int64_t output_size,414const int64_t index_size,415const int64_t data_size,416const float* input,417const int* indices,418const int* offsets,419const float* weights,420const float* scale_bias,421bool normalize_by_lengths,422float* out) {423return EmbeddingLookupIdx_int32_t_float_float__avx2_fma<false>(424block_size,425output_size,426index_size,427data_size,428input,429indices,430offsets,431weights,432scale_bias,433normalize_by_lengths,434out);435}436bool EmbeddingLookupIdx_int32_t_float_float_true__avx2_fma(437const int64_t block_size,438const int64_t output_size,439const int64_t index_size,440const int64_t data_size,441const float* input,442const int* indices,443const int* offsets,444const float* weights,445const float* scale_bias,446bool normalize_by_lengths,447float* out) {448return EmbeddingLookupIdx_int32_t_float_float__avx2_fma<true>(449block_size,450output_size,451index_size,452data_size,453input,454indices,455offsets,456weights,457scale_bias,458normalize_by_lengths,459out);460}461}860bool EmbeddingLookupIdx_int64_t_float_float_false__avx2_fma(861const int64_t block_size,862const int64_t output_size,863const int64_t index_size,864const int64_t data_size,865const float* input,866const int64_t* indices,867const int64_t* offsets,868const float* weights,869const float* scale_bias,870bool normalize_by_lengths,871float* out) {872return EmbeddingLookupIdx_int64_t_float_float__avx2_fma<false>(873block_size,874output_size,875index_size,876data_size,877input,878indices,879offsets,880weights,881scale_bias,882normalize_by_lengths,883out);884}885bool EmbeddingLookupIdx_int64_t_float_float_true__avx2_fma(886const int64_t block_size,887const int64_t output_size,888const int64_t index_size,889const int64_t data_size,890const float* input,891const int64_t* indices,892const int64_t* offsets,893const float* weights,894const float* scale_bias,895bool normalize_by_lengths,896float* out) {897return EmbeddingLookupIdx_int64_t_float_float__avx2_fma<true>(898block_size,899output_size,900index_size,901data_size,902input,903indices,904offsets,905weights,906scale_bias,907normalize_by_lengths,908out);909}910}1429bool EmbeddingLookupIdx_int32_t_half_float_false__avx2_fma(1430const int64_t block_size,1431const int64_t output_size,1432const int64_t index_size,1433const int64_t data_size,1434const at::Half* input,1435const int* indices,1436const int* offsets,1437const float* weights,1438const float* scale_bias,1439bool normalize_by_lengths,1440float* out) {1441return EmbeddingLookupIdx_int32_t_half_float__avx2_fma<false>(1442block_size,1443output_size,1444index_size,1445data_size,1446input,1447indices,1448offsets,1449weights,1450scale_bias,1451normalize_by_lengths,1452out);1453}1454bool EmbeddingLookupIdx_int32_t_half_float_true__avx2_fma(1455const int64_t block_size,1456const int64_t output_size,1457const int64_t index_size,1458const int64_t data_size,1459const at::Half* input,1460const int* indices,1461const int* offsets,1462const float* weights,1463const float* scale_bias,1464bool normalize_by_lengths,1465float* out) {1466return EmbeddingLookupIdx_int32_t_half_float__avx2_fma<true>(1467block_size,1468output_size,1469index_size,1470data_size,1471input,1472indices,1473offsets,1474weights,1475scale_bias,1476normalize_by_lengths,1477out);1478}1479}1998bool EmbeddingLookupIdx_int64_t_half_float_false__avx2_fma(1999const int64_t block_size,2000const int64_t output_size,2001const int64_t index_size,2002const int64_t data_size,2003const at::Half* input,2004const int64_t* indices,2005const int64_t* offsets,2006const float* weights,2007const float* scale_bias,2008bool normalize_by_lengths,2009float* out) {2010return EmbeddingLookupIdx_int64_t_half_float__avx2_fma<false>(2011block_size,2012output_size,2013index_size,2014data_size,2015input,2016indices,2017offsets,2018weights,2019scale_bias,2020normalize_by_lengths,2021out);2022}2023bool EmbeddingLookupIdx_int64_t_half_float_true__avx2_fma(2024const int64_t block_size,2025const int64_t output_size,2026const int64_t index_size,2027const int64_t data_size,2028const at::Half* input,2029const int64_t* indices,2030const int64_t* offsets,2031const float* weights,2032const float* scale_bias,2033bool normalize_by_lengths,2034float* out) {2035return EmbeddingLookupIdx_int64_t_half_float__avx2_fma<true>(2036block_size,2037output_size,2038index_size,2039data_size,2040input,2041indices,2042offsets,2043weights,2044scale_bias,2045normalize_by_lengths,2046out);2047}2048}2630bool EmbeddingLookupIdx_int32_t_bfloat16_float_false__avx2_fma(2631const int64_t block_size,2632const int64_t output_size,2633const int64_t index_size,2634const int64_t data_size,2635const at::BFloat16* input,2636const int* indices,2637const int* offsets,2638const float* weights,2639const float* scale_bias,2640bool normalize_by_lengths,2641float* out) {2642return EmbeddingLookupIdx_int32_t_bfloat16_float__avx2_fma<false>(2643block_size,2644output_size,2645index_size,2646data_size,2647input,2648indices,2649offsets,2650weights,2651scale_bias,2652normalize_by_lengths,2653out);2654}2655bool EmbeddingLookupIdx_int32_t_bfloat16_float_true__avx2_fma(2656const int64_t block_size,2657const int64_t output_size,2658const int64_t index_size,2659const int64_t data_size,2660const at::BFloat16* input,2661const int* indices,2662const int* offsets,2663const float* weights,2664const float* scale_bias,2665bool normalize_by_lengths,2666float* out) {2667return EmbeddingLookupIdx_int32_t_bfloat16_float__avx2_fma<true>(2668block_size,2669output_size,2670index_size,2671data_size,2672input,2673indices,2674offsets,2675weights,2676scale_bias,2677normalize_by_lengths,2678out);2679}2680}3262bool EmbeddingLookupIdx_int64_t_bfloat16_float_false__avx2_fma(3263const int64_t block_size,3264const int64_t output_size,3265const int64_t index_size,3266const int64_t data_size,3267const at::BFloat16* input,3268const int64_t* indices,3269const int64_t* offsets,3270const float* weights,3271const float* scale_bias,3272bool normalize_by_lengths,3273float* out) {3274return EmbeddingLookupIdx_int64_t_bfloat16_float__avx2_fma<false>(3275block_size,3276output_size,3277index_size,3278data_size,3279input,3280indices,3281offsets,3282weights,3283scale_bias,3284normalize_by_lengths,3285out);3286}3287bool EmbeddingLookupIdx_int64_t_bfloat16_float_true__avx2_fma(3288const int64_t block_size,3289const int64_t output_size,3290const int64_t index_size,3291const int64_t data_size,3292const at::BFloat16* input,3293const int64_t* indices,3294const int64_t* offsets,3295const float* weights,3296const float* scale_bias,3297bool normalize_by_lengths,3298float* out) {3299return EmbeddingLookupIdx_int64_t_bfloat16_float__avx2_fma<true>(3300block_size,3301output_size,3302index_size,3303data_size,3304input,3305indices,3306offsets,3307weights,3308scale_bias,3309normalize_by_lengths,3310out);3311}3312}3849bool EmbeddingLookupIdx_int32_t_uint8_t_float_false__avx2_fma(3850const int64_t block_size,3851const int64_t output_size,3852const int64_t index_size,3853const int64_t data_size,3854const uint8_t* input,3855const int* indices,3856const int* offsets,3857const float* weights,3858const float* scale_bias,3859bool normalize_by_lengths,3860float* out) {3861return EmbeddingLookupIdx_int32_t_uint8_t_float__avx2_fma<false>(3862block_size,3863output_size,3864index_size,3865data_size,3866input,3867indices,3868offsets,3869weights,3870scale_bias,3871normalize_by_lengths,3872out);3873}3874bool EmbeddingLookupIdx_int32_t_uint8_t_float_true__avx2_fma(3875const int64_t block_size,3876const int64_t output_size,3877const int64_t index_size,3878const int64_t data_size,3879const uint8_t* input,3880const int* indices,3881const int* offsets,3882const float* weights,3883const float* scale_bias,3884bool normalize_by_lengths,3885float* out) {3886return EmbeddingLookupIdx_int32_t_uint8_t_float__avx2_fma<true>(3887block_size,3888output_size,3889index_size,3890data_size,3891input,3892indices,3893offsets,3894weights,3895scale_bias,3896normalize_by_lengths,3897out);3898}3899}4436bool EmbeddingLookupIdx_int64_t_uint8_t_float_false__avx2_fma(4437const int64_t block_size,4438const int64_t output_size,4439const int64_t index_size,4440const int64_t data_size,4441const uint8_t* input,4442const int64_t* indices,4443const int64_t* offsets,4444const float* weights,4445const float* scale_bias,4446bool normalize_by_lengths,4447float* out) {4448return EmbeddingLookupIdx_int64_t_uint8_t_float__avx2_fma<false>(4449block_size,4450output_size,4451index_size,4452data_size,4453input,4454indices,4455offsets,4456weights,4457scale_bias,4458normalize_by_lengths,4459out);4460}4461bool EmbeddingLookupIdx_int64_t_uint8_t_float_true__avx2_fma(4462const int64_t block_size,4463const int64_t output_size,4464const int64_t index_size,4465const int64_t data_size,4466const uint8_t* input,4467const int64_t* indices,4468const int64_t* offsets,4469const float* weights,4470const float* scale_bias,4471bool normalize_by_lengths,4472float* out) {4473return EmbeddingLookupIdx_int64_t_uint8_t_float__avx2_fma<true>(4474block_size,4475output_size,4476index_size,4477data_size,4478input,4479indices,4480offsets,4481weights,4482scale_bias,4483normalize_by_lengths,4484out);4485}4486