main

BranchesTags

Go to file

Code

Open more actions menu

Folders and files

Name		Name	Last commit message	Last commit date
Latest commit   History 339 Commits 339 Commits
mse_loss		mse_loss
run_shell		run_shell
test__c_concat		test__c_concat
test__c_identity		test__c_identity
test_adamw		test_adamw
test_add		test_add
test_add_inplace		test_add_inplace
test_all		test_all
test_arange		test_arange
test_assign		test_assign
test_bitwise_not		test_bitwise_not
test_bmm		test_bmm
test_broadcast		test_broadcast
test_cast		test_cast
test_clip		test_clip
test_concat		test_concat
test_conv_nd		test_conv_nd
test_cos		test_cos
test_cross_entropy		test_cross_entropy
test_cumprod		test_cumprod
test_dist_split		test_dist_split
test_divide		test_divide
test_einsum		test_einsum
test_embedding		test_embedding
test_empty		test_empty
test_empty_like		test_empty_like
test_equal		test_equal
test_exp		test_exp
test_expand		test_expand
test_fill_		test_fill_
test_fill_constant		test_fill_constant
test_flash_attention		test_flash_attention
test_flatten		test_flatten
test_fp32_with_bf16		test_fp32_with_bf16
test_full		test_full
test_full_like		test_full_like
test_fused_linear		test_fused_linear
test_fused_linear_param_grad_add		test_fused_linear_param_grad_add
test_gather		test_gather
test_gaussian		test_gaussian
test_gelu		test_gelu
test_greater_than		test_greater_than
test_group_norm		test_group_norm
test_layernorm		test_layernorm
test_linear		test_linear
test_linspace		test_linspace
test_matmul		test_matmul
test_max		test_max
test_maximum		test_maximum
test_mean		test_mean
test_memory_efficient_attention		test_memory_efficient_attention
test_mp_allreduce		test_mp_allreduce
test_multiply		test_multiply
test_norm		test_norm
test_not_equal		test_not_equal
test_numel		test_numel
test_pad		test_pad
test_parallel_cross_entropy		test_parallel_cross_entropy
test_pow		test_pow
test_randint		test_randint
test_reshape		test_reshape
test_reshape_inplace		test_reshape_inplace
test_rsqrt		test_rsqrt
test_scale		test_scale
test_scaled_dot_product_attention		test_scaled_dot_product_attention
test_scatter		test_scatter
test_scatter_nd_add		test_scatter_nd_add
test_shape		test_shape
test_silu		test_silu
test_sin		test_sin
test_slice		test_slice
test_softmax		test_softmax
test_split		test_split
test_sqrt		test_sqrt
test_square		test_square
test_squeeze		test_squeeze
test_stack		test_stack
test_subtract		test_subtract
test_sum		test_sum
test_tile		test_tile
test_transpose		test_transpose
test_uniform		test_uniform
test_unsqueeze		test_unsqueeze
test_vocab_parallel_embedding		test_vocab_parallel_embedding
test_where		test_where
test_zero_		test_zero_
api_test.py		api_test.py
utils.py		utils.py
View all files

About

No description, website, or topics provided.

Activity

Stars

0 stars

Watchers

1 watching

Forks

25 forks

Report repository

Releases

Packages

Contributors

Languages