[package]
edition = "2024"
rust-version = "1.85"
name = "baracuda-kernels"
version = "0.0.1-alpha.68"
authors = ["The baracuda contributors"]
build = false
exclude = [
"*.profraw",
"target/",
]
autolib = false
autobins = false
autoexamples = false
autotests = false
autobenches = false
description = "Unified ML op facade for the baracuda CUDA ecosystem. Exposes every primitive an ML framework would expect (union of PyTorch torch.* + nn.functional and JAX lax.* / numpy ops) through a single Plan-based Rust surface, internally dispatching to baracuda-cutlass, the baracuda-* NVIDIA-library wrappers, or bespoke baracuda-kernels-sys kernels."
homepage = "https://github.com/ciresnave/baracuda"
readme = "README.md"
keywords = [
"cuda",
"gpu",
"ml",
"kernel",
"pytorch",
]
categories = [
"api-bindings",
"science",
]
license = "MIT OR Apache-2.0"
repository = "https://github.com/ciresnave/baracuda"
[package.metadata.docs.rs]
features = ["sm80"]
no-default-features = false
[features]
awq = ["baracuda-kernels-sys/awq"]
bnb_nf4 = ["baracuda-kernels-sys/bnb_nf4"]
cudnn = ["baracuda-kernels-sys/cudnn"]
default = [
"sm80",
"fa2",
]
fa2 = ["baracuda-kernels-sys/fa2"]
flashinfer = ["baracuda-kernels-sys/flashinfer"]
mamba = ["baracuda-kernels-sys/mamba"]
marlin = ["baracuda-kernels-sys/marlin"]
megatron_tp = [
"dep:baracuda-megatron",
"baracuda-megatron/half-crate",
]
mhc = ["baracuda-kernels-sys/mhc"]
nvshmem = [
"dep:baracuda-nvshmem",
"dep:baracuda-nvshmem-sys",
]
optim = [
"dep:baracuda-optim",
"baracuda-optim/sm80",
]
ozimmu = ["baracuda-cutlass/ozimmu"]
ring_attention = [
"baracuda-kernels-sys/ring_attention",
"dep:baracuda-nccl",
"dep:baracuda-nccl-sys",
]
sm80 = [
"baracuda-kernels-sys/sm80",
"baracuda-cutlass/sm80",
]
sm89 = ["baracuda-kernels-sys/sm89"]
sm90a = [
"baracuda-kernels-sys/sm90a",
"baracuda-cutlass/sm90a",
]
tensor_engine = [
"dep:baracuda-transformer-engine",
"baracuda-transformer-engine/sm80",
]
xformers_blocksparse = ["baracuda-kernels-sys/xformers_blocksparse"]
xformers_sparse24 = ["baracuda-kernels-sys/xformers_sparse24"]
[lib]
name = "baracuda_kernels"
path = "src/lib.rs"
[[example]]
name = "speculative_decode_compose"
path = "examples/speculative_decode_compose.rs"
[[test]]
name = "adaptive_avg_pool2d_smoke"
path = "tests/adaptive_avg_pool2d_smoke.rs"
[[test]]
name = "adaptive_pool_bitexact_smoke"
path = "tests/adaptive_pool_bitexact_smoke.rs"
[[test]]
name = "affine_inplace_ffi_smoke"
path = "tests/affine_inplace_ffi_smoke.rs"
[[test]]
name = "affine_smoke"
path = "tests/affine_smoke.rs"
[[test]]
name = "affine_strided_smoke"
path = "tests/affine_strided_smoke.rs"
[[test]]
name = "alibi_backward_smoke"
path = "tests/alibi_backward_smoke.rs"
[[test]]
name = "alibi_smoke"
path = "tests/alibi_smoke.rs"
[[test]]
name = "arg_axis_dtype_smoke"
path = "tests/arg_axis_dtype_smoke.rs"
[[test]]
name = "arg_reduce_dtype_smoke"
path = "tests/arg_reduce_dtype_smoke.rs"
[[test]]
name = "arg_reduce_int_dtype_smoke"
path = "tests/arg_reduce_int_dtype_smoke.rs"
[[test]]
name = "arg_reduce_smoke"
path = "tests/arg_reduce_smoke.rs"
[[test]]
name = "argsort_big_smoke"
path = "tests/argsort_big_smoke.rs"
[[test]]
name = "argsort_dtype_extras_smoke"
path = "tests/argsort_dtype_extras_smoke.rs"
[[test]]
name = "argsort_smoke"
path = "tests/argsort_smoke.rs"
[[test]]
name = "avg_pool1d_smoke"
path = "tests/avg_pool1d_smoke.rs"
[[test]]
name = "avg_pool3d_smoke"
path = "tests/avg_pool3d_smoke.rs"
[[test]]
name = "awq_smoke"
path = "tests/awq_smoke.rs"
[[test]]
name = "batch_norm_backward_smoke"
path = "tests/batch_norm_backward_smoke.rs"
[[test]]
name = "batch_norm_inplace_smoke"
path = "tests/batch_norm_inplace_smoke.rs"
[[test]]
name = "batch_norm_smoke"
path = "tests/batch_norm_smoke.rs"
[[test]]
name = "bin_rrr_smoke"
path = "tests/bin_rrr_smoke.rs"
[[test]]
name = "bin_smoke"
path = "tests/bin_smoke.rs"
[[test]]
name = "binary_add_backward_smoke"
path = "tests/binary_add_backward_smoke.rs"
[[test]]
name = "binary_add_smoke"
path = "tests/binary_add_smoke.rs"
[[test]]
name = "binary_atan2_backward_smoke"
path = "tests/binary_atan2_backward_smoke.rs"
[[test]]
name = "binary_atan2_smoke"
path = "tests/binary_atan2_smoke.rs"
[[test]]
name = "binary_bitwise_int_smoke"
path = "tests/binary_bitwise_int_smoke.rs"
[[test]]
name = "binary_broadcast_dtype_smoke"
path = "tests/binary_broadcast_dtype_smoke.rs"
[[test]]
name = "binary_broadcast_smoke"
path = "tests/binary_broadcast_smoke.rs"
[[test]]
name = "binary_cmp_eq_dtype_smoke"
path = "tests/binary_cmp_eq_dtype_smoke.rs"
[[test]]
name = "binary_cmp_eq_smoke"
path = "tests/binary_cmp_eq_smoke.rs"
[[test]]
name = "binary_cmp_ge_smoke"
path = "tests/binary_cmp_ge_smoke.rs"
[[test]]
name = "binary_cmp_gt_smoke"
path = "tests/binary_cmp_gt_smoke.rs"
[[test]]
name = "binary_cmp_le_smoke"
path = "tests/binary_cmp_le_smoke.rs"
[[test]]
name = "binary_cmp_lt_smoke"
path = "tests/binary_cmp_lt_smoke.rs"
[[test]]
name = "binary_cmp_ne_smoke"
path = "tests/binary_cmp_ne_smoke.rs"
[[test]]
name = "binary_copysign_smoke"
path = "tests/binary_copysign_smoke.rs"
[[test]]
name = "binary_div_backward_smoke"
path = "tests/binary_div_backward_smoke.rs"
[[test]]
name = "binary_div_smoke"
path = "tests/binary_div_smoke.rs"
[[test]]
name = "binary_floor_divide_smoke"
path = "tests/binary_floor_divide_smoke.rs"
[[test]]
name = "binary_fmax_smoke"
path = "tests/binary_fmax_smoke.rs"
[[test]]
name = "binary_fmin_smoke"
path = "tests/binary_fmin_smoke.rs"
[[test]]
name = "binary_hypot_backward_smoke"
path = "tests/binary_hypot_backward_smoke.rs"
[[test]]
name = "binary_hypot_smoke"
path = "tests/binary_hypot_smoke.rs"
[[test]]
name = "binary_lerp_backward_smoke"
path = "tests/binary_lerp_backward_smoke.rs"
[[test]]
name = "binary_lerp_smoke"
path = "tests/binary_lerp_smoke.rs"
[[test]]
name = "binary_logical_bool_smoke"
path = "tests/binary_logical_bool_smoke.rs"
[[test]]
name = "binary_maximum_backward_smoke"
path = "tests/binary_maximum_backward_smoke.rs"
[[test]]
name = "binary_maximum_smoke"
path = "tests/binary_maximum_smoke.rs"
[[test]]
name = "binary_minimum_backward_smoke"
path = "tests/binary_minimum_backward_smoke.rs"
[[test]]
name = "binary_minimum_smoke"
path = "tests/binary_minimum_smoke.rs"
[[test]]
name = "binary_mod_smoke"
path = "tests/binary_mod_smoke.rs"
[[test]]
name = "binary_mul_backward_smoke"
path = "tests/binary_mul_backward_smoke.rs"
[[test]]
name = "binary_mul_smoke"
path = "tests/binary_mul_smoke.rs"
[[test]]
name = "binary_nextafter_smoke"
path = "tests/binary_nextafter_smoke.rs"
[[test]]
name = "binary_pow_backward_smoke"
path = "tests/binary_pow_backward_smoke.rs"
[[test]]
name = "binary_pow_smoke"
path = "tests/binary_pow_smoke.rs"
[[test]]
name = "binary_remainder_smoke"
path = "tests/binary_remainder_smoke.rs"
[[test]]
name = "binary_sub_backward_smoke"
path = "tests/binary_sub_backward_smoke.rs"
[[test]]
name = "binary_sub_smoke"
path = "tests/binary_sub_smoke.rs"
[[test]]
name = "bincount_smoke"
path = "tests/bincount_smoke.rs"
[[test]]
name = "cascade_attn_smoke"
path = "tests/cascade_attn_smoke.rs"
[[test]]
name = "cast_smoke"
path = "tests/cast_smoke.rs"
[[test]]
name = "cast_subbyte_smoke"
path = "tests/cast_subbyte_smoke.rs"
[[test]]
name = "cast_u32_i16_smoke"
path = "tests/cast_u32_i16_smoke.rs"
[[test]]
name = "causal_conv1d_bw_smoke"
path = "tests/causal_conv1d_bw_smoke.rs"
[[test]]
name = "causal_conv1d_smoke"
path = "tests/causal_conv1d_smoke.rs"
[[test]]
name = "concat_backward_smoke"
path = "tests/concat_backward_smoke.rs"
[[test]]
name = "concat_dtype_smoke"
path = "tests/concat_dtype_smoke.rs"
[[test]]
name = "concat_smoke"
path = "tests/concat_smoke.rs"
[[test]]
name = "contiguize_smoke"
path = "tests/contiguize_smoke.rs"
[[test]]
name = "conv1d_smoke"
path = "tests/conv1d_smoke.rs"
[[test]]
name = "conv2d_depthwise_smoke"
path = "tests/conv2d_depthwise_smoke.rs"
[[test]]
name = "conv2d_smoke"
path = "tests/conv2d_smoke.rs"
[[test]]
name = "conv3d_smoke"
path = "tests/conv3d_smoke.rs"
[[test]]
name = "conv_cudnn_ffi_smoke"
path = "tests/conv_cudnn_ffi_smoke.rs"
[[test]]
name = "conv_transpose2d_smoke"
path = "tests/conv_transpose2d_smoke.rs"
[[test]]
name = "ctc_loss_cudnn_smoke"
path = "tests/ctc_loss_cudnn_smoke.rs"
[[test]]
name = "cufft_ffi_smoke"
path = "tests/cufft_ffi_smoke.rs"
[[test]]
name = "curand_ffi_smoke"
path = "tests/curand_ffi_smoke.rs"
[[test]]
name = "cusolver_ffi_smoke"
path = "tests/cusolver_ffi_smoke.rs"
[[test]]
name = "cutlass_ffi_smoke"
path = "tests/cutlass_ffi_smoke.rs"
[[test]]
name = "dense_gemm_smoke"
path = "tests/dense_gemm_smoke.rs"
[[test]]
name = "dequantize_per_group_smoke"
path = "tests/dequantize_per_group_smoke.rs"
[[test]]
name = "dequantize_per_tensor_smoke"
path = "tests/dequantize_per_tensor_smoke.rs"
[[test]]
name = "dequantize_per_token_smoke"
path = "tests/dequantize_per_token_smoke.rs"
[[test]]
name = "dynamic_range_quantize_smoke"
path = "tests/dynamic_range_quantize_smoke.rs"
[[test]]
name = "embedding_backward_smoke"
path = "tests/embedding_backward_smoke.rs"
[[test]]
name = "embedding_bag_backward_smoke"
path = "tests/embedding_bag_backward_smoke.rs"
[[test]]
name = "embedding_bag_max_backward_smoke"
path = "tests/embedding_bag_max_backward_smoke.rs"
[[test]]
name = "embedding_bag_max_smoke"
path = "tests/embedding_bag_max_smoke.rs"
[[test]]
name = "embedding_bag_smoke"
path = "tests/embedding_bag_smoke.rs"
[[test]]
name = "embedding_smoke"
path = "tests/embedding_smoke.rs"
[[test]]
name = "fa2_alibi_smoke"
path = "tests/fa2_alibi_smoke.rs"
[[test]]
name = "fa2_arbmask_smoke"
path = "tests/fa2_arbmask_smoke.rs"
[[test]]
name = "fa2_backward_smoke"
path = "tests/fa2_backward_smoke.rs"
[[test]]
name = "fa2_fw_bw_saved_tensor_smoke"
path = "tests/fa2_fw_bw_saved_tensor_smoke.rs"
[[test]]
name = "fa2_gqa_smoke"
path = "tests/fa2_gqa_smoke.rs"
[[test]]
name = "fa2_hdim_fanout_smoke"
path = "tests/fa2_hdim_fanout_smoke.rs"
[[test]]
name = "fa2_sliding_window_smoke"
path = "tests/fa2_sliding_window_smoke.rs"
[[test]]
name = "fa2_smoke"
path = "tests/fa2_smoke.rs"
[[test]]
name = "fa2_softcap_smoke"
path = "tests/fa2_softcap_smoke.rs"
[[test]]
name = "fa2_varlen_smoke"
path = "tests/fa2_varlen_smoke.rs"
[[test]]
name = "fake_quantize_smoke"
path = "tests/fake_quantize_smoke.rs"
[[test]]
name = "fft_2d_smoke"
path = "tests/fft_2d_smoke.rs"
[[test]]
name = "fft_3d_smoke"
path = "tests/fft_3d_smoke.rs"
[[test]]
name = "fft_fft_smoke"
path = "tests/fft_fft_smoke.rs"
[[test]]
name = "fft_rfft_smoke"
path = "tests/fft_rfft_smoke.rs"
[[test]]
name = "fft_shift_smoke"
path = "tests/fft_shift_smoke.rs"
[[test]]
name = "fftshift_nd_smoke"
path = "tests/fftshift_nd_smoke.rs"
[[test]]
name = "fill_extras_smoke"
path = "tests/fill_extras_smoke.rs"
[[test]]
name = "fill_smoke"
path = "tests/fill_smoke.rs"
[[test]]
name = "flash_decoding_smoke"
path = "tests/flash_decoding_smoke.rs"
[[test]]
name = "flash_sdpa_backward_smoke"
path = "tests/flash_sdpa_backward_smoke.rs"
[[test]]
name = "flash_sdpa_sm89_smoke"
path = "tests/flash_sdpa_sm89_smoke.rs"
[[test]]
name = "flash_sdpa_sm89_strided_smoke"
path = "tests/flash_sdpa_sm89_strided_smoke.rs"
[[test]]
name = "flash_sdpa_smoke"
path = "tests/flash_sdpa_smoke.rs"
[[test]]
name = "flip_backward_smoke"
path = "tests/flip_backward_smoke.rs"
[[test]]
name = "flip_dtype_smoke"
path = "tests/flip_dtype_smoke.rs"
[[test]]
name = "flip_smoke"
path = "tests/flip_smoke.rs"
[[test]]
name = "fp8_bias_smoke"
path = "tests/fp8_bias_smoke.rs"
[[test]]
name = "fp8_e4m3_rcr_smoke"
path = "tests/fp8_e4m3_rcr_smoke.rs"
[[test]]
name = "fp8_e4m3_rrr_smoke"
path = "tests/fp8_e4m3_rrr_smoke.rs"
[[test]]
name = "fp8_e5m2_rcr_smoke"
path = "tests/fp8_e5m2_rcr_smoke.rs"
[[test]]
name = "fp8_e5m2_rrr_smoke"
path = "tests/fp8_e5m2_rrr_smoke.rs"
[[test]]
name = "fractional_max_pool_smoke"
path = "tests/fractional_max_pool_smoke.rs"
[[test]]
name = "fused_linear_cross_entropy_bw_smoke"
path = "tests/fused_linear_cross_entropy_bw_smoke.rs"
[[test]]
name = "fused_linear_cross_entropy_memory_smoke"
path = "tests/fused_linear_cross_entropy_memory_smoke.rs"
[[test]]
name = "fused_linear_cross_entropy_smoke"
path = "tests/fused_linear_cross_entropy_smoke.rs"
[[test]]
name = "gated_geglu_backward_smoke"
path = "tests/gated_geglu_backward_smoke.rs"
[[test]]
name = "gated_geglu_smoke"
path = "tests/gated_geglu_smoke.rs"
[[test]]
name = "gated_glu_backward_smoke"
path = "tests/gated_glu_backward_smoke.rs"
[[test]]
name = "gated_glu_smoke"
path = "tests/gated_glu_smoke.rs"
[[test]]
name = "gated_reglu_backward_smoke"
path = "tests/gated_reglu_backward_smoke.rs"
[[test]]
name = "gated_reglu_smoke"
path = "tests/gated_reglu_smoke.rs"
[[test]]
name = "gated_swiglu_backward_smoke"
path = "tests/gated_swiglu_backward_smoke.rs"
[[test]]
name = "gated_swiglu_smoke"
path = "tests/gated_swiglu_smoke.rs"
[[test]]
name = "gather_backward_smoke"
path = "tests/gather_backward_smoke.rs"
[[test]]
name = "gather_idx_extras_smoke"
path = "tests/gather_idx_extras_smoke.rs"
[[test]]
name = "gather_smoke"
path = "tests/gather_smoke.rs"
[[test]]
name = "gemm_sparse24_smoke"
path = "tests/gemm_sparse24_smoke.rs"
[[test]]
name = "gguf_dequantize_smoke"
path = "tests/gguf_dequantize_smoke.rs"
[[test]]
name = "gguf_mmvq_smoke"
path = "tests/gguf_mmvq_smoke.rs"
[[test]]
name = "gptq_to_marlin_smoke"
path = "tests/gptq_to_marlin_smoke.rs"
[[test]]
name = "grid_sample_smoke"
path = "tests/grid_sample_smoke.rs"
[[test]]
name = "group_norm_backward_smoke"
path = "tests/group_norm_backward_smoke.rs"
[[test]]
name = "group_norm_inplace_smoke"
path = "tests/group_norm_inplace_smoke.rs"
[[test]]
name = "group_norm_smoke"
path = "tests/group_norm_smoke.rs"
[[test]]
name = "gumbel_softmax_backward_smoke"
path = "tests/gumbel_softmax_backward_smoke.rs"
[[test]]
name = "gumbel_softmax_smoke"
path = "tests/gumbel_softmax_smoke.rs"
[[test]]
name = "histogram_smoke"
path = "tests/histogram_smoke.rs"
[[test]]
name = "hyper_connection_smoke"
path = "tests/hyper_connection_smoke.rs"
[[test]]
name = "im2col_smoke"
path = "tests/im2col_smoke.rs"
[[test]]
name = "index_add_smoke"
path = "tests/index_add_smoke.rs"
[[test]]
name = "index_select_backward_smoke"
path = "tests/index_select_backward_smoke.rs"
[[test]]
name = "index_select_smoke"
path = "tests/index_select_smoke.rs"
[[test]]
name = "indexing_int_dtype_smoke"
path = "tests/indexing_int_dtype_smoke.rs"
[[test]]
name = "inplace_aliasing_contract_smoke"
path = "tests/inplace_aliasing_contract_smoke.rs"
[[test]]
name = "inplace_aliasing_extended_smoke"
path = "tests/inplace_aliasing_extended_smoke.rs"
[[test]]
name = "instance_norm_backward_smoke"
path = "tests/instance_norm_backward_smoke.rs"
[[test]]
name = "instance_norm_inplace_smoke"
path = "tests/instance_norm_inplace_smoke.rs"
[[test]]
name = "instance_norm_smoke"
path = "tests/instance_norm_smoke.rs"
[[test]]
name = "int4_bias_smoke"
path = "tests/int4_bias_smoke.rs"
[[test]]
name = "int4_rrr_smoke"
path = "tests/int4_rrr_smoke.rs"
[[test]]
name = "int8_rrr_smoke"
path = "tests/int8_rrr_smoke.rs"
[[test]]
name = "interpolate_smoke"
path = "tests/interpolate_smoke.rs"
[[test]]
name = "kv_cache_smoke"
path = "tests/kv_cache_smoke.rs"
[[test]]
name = "layer_norm_backward_smoke"
path = "tests/layer_norm_backward_smoke.rs"
[[test]]
name = "layer_norm_inplace_smoke"
path = "tests/layer_norm_inplace_smoke.rs"
[[test]]
name = "layer_norm_smoke"
path = "tests/layer_norm_smoke.rs"
[[test]]
name = "linalg_cholesky_smoke"
path = "tests/linalg_cholesky_smoke.rs"
[[test]]
name = "linalg_eig_smoke"
path = "tests/linalg_eig_smoke.rs"
[[test]]
name = "linalg_eigh_smoke"
path = "tests/linalg_eigh_smoke.rs"
[[test]]
name = "linalg_inverse_smoke"
path = "tests/linalg_inverse_smoke.rs"
[[test]]
name = "linalg_lstsq_smoke"
path = "tests/linalg_lstsq_smoke.rs"
[[test]]
name = "linalg_lu_smoke"
path = "tests/linalg_lu_smoke.rs"
[[test]]
name = "linalg_ormqr_batched_smoke"
path = "tests/linalg_ormqr_batched_smoke.rs"
[[test]]
name = "linalg_ormqr_batched_wy_smoke"
path = "tests/linalg_ormqr_batched_wy_smoke.rs"
[[test]]
name = "linalg_qr_batched_materialize_smoke"
path = "tests/linalg_qr_batched_materialize_smoke.rs"
[[test]]
name = "linalg_qr_batched_smoke"
path = "tests/linalg_qr_batched_smoke.rs"
[[test]]
name = "linalg_qr_smoke"
path = "tests/linalg_qr_smoke.rs"
[[test]]
name = "linalg_solve_smoke"
path = "tests/linalg_solve_smoke.rs"
[[test]]
name = "linalg_svd_batched_smoke"
path = "tests/linalg_svd_batched_smoke.rs"
[[test]]
name = "linalg_svd_smoke"
path = "tests/linalg_svd_smoke.rs"
[[test]]
name = "linalg_svda_batched_smoke"
path = "tests/linalg_svda_batched_smoke.rs"
[[test]]
name = "log_softmax_backward_smoke"
path = "tests/log_softmax_backward_smoke.rs"
[[test]]
name = "log_softmax_inplace_smoke"
path = "tests/log_softmax_inplace_smoke.rs"
[[test]]
name = "log_softmax_smoke"
path = "tests/log_softmax_smoke.rs"
[[test]]
name = "loss_bce_backward_smoke"
path = "tests/loss_bce_backward_smoke.rs"
[[test]]
name = "loss_bce_smoke"
path = "tests/loss_bce_smoke.rs"
[[test]]
name = "loss_bce_with_logits_backward_smoke"
path = "tests/loss_bce_with_logits_backward_smoke.rs"
[[test]]
name = "loss_bce_with_logits_smoke"
path = "tests/loss_bce_with_logits_smoke.rs"
[[test]]
name = "loss_cosine_embedding_backward_smoke"
path = "tests/loss_cosine_embedding_backward_smoke.rs"
[[test]]
name = "loss_cosine_embedding_smoke"
path = "tests/loss_cosine_embedding_smoke.rs"
[[test]]
name = "loss_cross_entropy_backward_smoke"
path = "tests/loss_cross_entropy_backward_smoke.rs"
[[test]]
name = "loss_cross_entropy_smoke"
path = "tests/loss_cross_entropy_smoke.rs"
[[test]]
name = "loss_cross_entropy_soft_backward_smoke"
path = "tests/loss_cross_entropy_soft_backward_smoke.rs"
[[test]]
name = "loss_cross_entropy_soft_smoke"
path = "tests/loss_cross_entropy_soft_smoke.rs"
[[test]]
name = "loss_ctc_backward_smoke"
path = "tests/loss_ctc_backward_smoke.rs"
[[test]]
name = "loss_ctc_smoke"
path = "tests/loss_ctc_smoke.rs"
[[test]]
name = "loss_gaussian_nll_backward_smoke"
path = "tests/loss_gaussian_nll_backward_smoke.rs"
[[test]]
name = "loss_gaussian_nll_smoke"
path = "tests/loss_gaussian_nll_smoke.rs"
[[test]]
name = "loss_hinge_embedding_backward_smoke"
path = "tests/loss_hinge_embedding_backward_smoke.rs"
[[test]]
name = "loss_hinge_embedding_smoke"
path = "tests/loss_hinge_embedding_smoke.rs"
[[test]]
name = "loss_huber_backward_smoke"
path = "tests/loss_huber_backward_smoke.rs"
[[test]]
name = "loss_huber_smoke"
path = "tests/loss_huber_smoke.rs"
[[test]]
name = "loss_kl_div_backward_smoke"
path = "tests/loss_kl_div_backward_smoke.rs"
[[test]]
name = "loss_kl_div_smoke"
path = "tests/loss_kl_div_smoke.rs"
[[test]]
name = "loss_l1_backward_smoke"
path = "tests/loss_l1_backward_smoke.rs"
[[test]]
name = "loss_l1_smoke"
path = "tests/loss_l1_smoke.rs"
[[test]]
name = "loss_margin_ranking_backward_smoke"
path = "tests/loss_margin_ranking_backward_smoke.rs"
[[test]]
name = "loss_margin_ranking_smoke"
path = "tests/loss_margin_ranking_smoke.rs"
[[test]]
name = "loss_mse_backward_smoke"
path = "tests/loss_mse_backward_smoke.rs"
[[test]]
name = "loss_mse_smoke"
path = "tests/loss_mse_smoke.rs"
[[test]]
name = "loss_multi_margin_backward_smoke"
path = "tests/loss_multi_margin_backward_smoke.rs"
[[test]]
name = "loss_multi_margin_smoke"
path = "tests/loss_multi_margin_smoke.rs"
[[test]]
name = "loss_multilabel_margin_backward_smoke"
path = "tests/loss_multilabel_margin_backward_smoke.rs"
[[test]]
name = "loss_multilabel_margin_smoke"
path = "tests/loss_multilabel_margin_smoke.rs"
[[test]]
name = "loss_multilabel_soft_margin_backward_smoke"
path = "tests/loss_multilabel_soft_margin_backward_smoke.rs"
[[test]]
name = "loss_multilabel_soft_margin_smoke"
path = "tests/loss_multilabel_soft_margin_smoke.rs"
[[test]]
name = "loss_nll_backward_smoke"
path = "tests/loss_nll_backward_smoke.rs"
[[test]]
name = "loss_nll_smoke"
path = "tests/loss_nll_smoke.rs"
[[test]]
name = "loss_poisson_nll_backward_smoke"
path = "tests/loss_poisson_nll_backward_smoke.rs"
[[test]]
name = "loss_poisson_nll_smoke"
path = "tests/loss_poisson_nll_smoke.rs"
[[test]]
name = "loss_smooth_l1_backward_smoke"
path = "tests/loss_smooth_l1_backward_smoke.rs"
[[test]]
name = "loss_smooth_l1_smoke"
path = "tests/loss_smooth_l1_smoke.rs"
[[test]]
name = "loss_triplet_margin_backward_smoke"
path = "tests/loss_triplet_margin_backward_smoke.rs"
[[test]]
name = "loss_triplet_margin_smoke"
path = "tests/loss_triplet_margin_smoke.rs"
[[test]]
name = "lp_pool_smoke"
path = "tests/lp_pool_smoke.rs"
[[test]]
name = "mamba1_block_smoke"
path = "tests/mamba1_block_smoke.rs"
[[test]]
name = "mamba2_block_smoke"
path = "tests/mamba2_block_smoke.rs"
[[test]]
name = "marlin_smoke"
path = "tests/marlin_smoke.rs"
[[test]]
name = "masked_fill_backward_smoke"
path = "tests/masked_fill_backward_smoke.rs"
[[test]]
name = "masked_fill_smoke"
path = "tests/masked_fill_smoke.rs"
[[test]]
name = "max_pool1d_smoke"
path = "tests/max_pool1d_smoke.rs"
[[test]]
name = "max_pool3d_smoke"
path = "tests/max_pool3d_smoke.rs"
[[test]]
name = "mmvq_batched_smoke"
path = "tests/mmvq_batched_smoke.rs"
[[test]]
name = "mmvq_f16_bf16_smoke"
path = "tests/mmvq_f16_bf16_smoke.rs"
[[test]]
name = "mmvq_multim_smoke"
path = "tests/mmvq_multim_smoke.rs"
[[test]]
name = "mmvq_strided_smoke"
path = "tests/mmvq_strided_smoke.rs"
[[test]]
name = "moe_ffi_direct_smoke"
path = "tests/moe_ffi_direct_smoke.rs"
[[test]]
name = "moe_gguf_smoke"
path = "tests/moe_gguf_smoke.rs"
[[test]]
name = "moe_wmma_gguf_smoke"
path = "tests/moe_wmma_gguf_smoke.rs"
[[test]]
name = "moe_wmma_smoke"
path = "tests/moe_wmma_smoke.rs"
[[test]]
name = "msort_smoke"
path = "tests/msort_smoke.rs"
[[test]]
name = "nf4_dequant_smoke"
path = "tests/nf4_dequant_smoke.rs"
[[test]]
name = "nf4_gemv_smoke"
path = "tests/nf4_gemv_smoke.rs"
[[test]]
name = "nf4_multim_smoke"
path = "tests/nf4_multim_smoke.rs"
[[test]]
name = "nms_smoke"
path = "tests/nms_smoke.rs"
[[test]]
name = "nonzero_i64idx_smoke"
path = "tests/nonzero_i64idx_smoke.rs"
[[test]]
name = "nonzero_smoke"
path = "tests/nonzero_smoke.rs"
[[test]]
name = "one_hot_i64idx_smoke"
path = "tests/one_hot_i64idx_smoke.rs"
[[test]]
name = "one_hot_smoke"
path = "tests/one_hot_smoke.rs"
[[test]]
name = "pad_backward_smoke"
path = "tests/pad_backward_smoke.rs"
[[test]]
name = "pad_dtype_smoke"
path = "tests/pad_dtype_smoke.rs"
[[test]]
name = "pad_modes_smoke"
path = "tests/pad_modes_smoke.rs"
[[test]]
name = "pad_smoke"
path = "tests/pad_smoke.rs"
[[test]]
name = "paged_decode_smoke"
path = "tests/paged_decode_smoke.rs"
[[test]]
name = "paged_kv_append_smoke"
path = "tests/paged_kv_append_smoke.rs"
[[test]]
name = "permute_backward_smoke"
path = "tests/permute_backward_smoke.rs"
[[test]]
name = "permute_dtype_smoke"
path = "tests/permute_dtype_smoke.rs"
[[test]]
name = "permute_smoke"
path = "tests/permute_smoke.rs"
[[test]]
name = "pixel_shuffle_smoke"
path = "tests/pixel_shuffle_smoke.rs"
[[test]]
name = "pool2d_smoke"
path = "tests/pool2d_smoke.rs"
[[test]]
name = "pool_cudnn_ffi_smoke"
path = "tests/pool_cudnn_ffi_smoke.rs"
[[test]]
name = "pool_extras_smoke"
path = "tests/pool_extras_smoke.rs"
[[test]]
name = "powi_strided_smoke"
path = "tests/powi_strided_smoke.rs"
[[test]]
name = "prelu_backward_smoke"
path = "tests/prelu_backward_smoke.rs"
[[test]]
name = "prelu_smoke"
path = "tests/prelu_smoke.rs"
[[test]]
name = "quantize_per_channel_smoke"
path = "tests/quantize_per_channel_smoke.rs"
[[test]]
name = "quantize_per_group_backward_smoke"
path = "tests/quantize_per_group_backward_smoke.rs"
[[test]]
name = "quantize_per_group_smoke"
path = "tests/quantize_per_group_smoke.rs"
[[test]]
name = "quantize_per_tensor_backward_smoke"
path = "tests/quantize_per_tensor_backward_smoke.rs"
[[test]]
name = "quantize_per_tensor_smoke"
path = "tests/quantize_per_tensor_smoke.rs"
[[test]]
name = "quantize_per_token_backward_smoke"
path = "tests/quantize_per_token_backward_smoke.rs"
[[test]]
name = "quantize_per_token_smoke"
path = "tests/quantize_per_token_smoke.rs"
[[test]]
name = "quantized_linear_smoke"
path = "tests/quantized_linear_smoke.rs"
[[test]]
name = "random_bernoulli_smoke"
path = "tests/random_bernoulli_smoke.rs"
[[test]]
name = "random_dropout_backward_smoke"
path = "tests/random_dropout_backward_smoke.rs"
[[test]]
name = "random_dropout_smoke"
path = "tests/random_dropout_smoke.rs"
[[test]]
name = "random_normal_smoke"
path = "tests/random_normal_smoke.rs"
[[test]]
name = "random_uniform_smoke"
path = "tests/random_uniform_smoke.rs"
[[test]]
name = "reduce_all_smoke"
path = "tests/reduce_all_smoke.rs"
[[test]]
name = "reduce_any_smoke"
path = "tests/reduce_any_smoke.rs"
[[test]]
name = "reduce_count_nonzero_smoke"
path = "tests/reduce_count_nonzero_smoke.rs"
[[test]]
name = "reduce_int_dtype_smoke"
path = "tests/reduce_int_dtype_smoke.rs"
[[test]]
name = "reduce_logsumexp_backward_smoke"
path = "tests/reduce_logsumexp_backward_smoke.rs"
[[test]]
name = "reduce_logsumexp_smoke"
path = "tests/reduce_logsumexp_smoke.rs"
[[test]]
name = "reduce_max_backward_smoke"
path = "tests/reduce_max_backward_smoke.rs"
[[test]]
name = "reduce_max_smoke"
path = "tests/reduce_max_smoke.rs"
[[test]]
name = "reduce_max_to_smoke"
path = "tests/reduce_max_to_smoke.rs"
[[test]]
name = "reduce_mean_backward_smoke"
path = "tests/reduce_mean_backward_smoke.rs"
[[test]]
name = "reduce_mean_smoke"
path = "tests/reduce_mean_smoke.rs"
[[test]]
name = "reduce_min_backward_smoke"
path = "tests/reduce_min_backward_smoke.rs"
[[test]]
name = "reduce_min_smoke"
path = "tests/reduce_min_smoke.rs"
[[test]]
name = "reduce_norm2_backward_smoke"
path = "tests/reduce_norm2_backward_smoke.rs"
[[test]]
name = "reduce_norm2_smoke"
path = "tests/reduce_norm2_smoke.rs"
[[test]]
name = "reduce_prod_backward_smoke"
path = "tests/reduce_prod_backward_smoke.rs"
[[test]]
name = "reduce_prod_smoke"
path = "tests/reduce_prod_smoke.rs"
[[test]]
name = "reduce_std_backward_smoke"
path = "tests/reduce_std_backward_smoke.rs"
[[test]]
name = "reduce_sum_backward_smoke"
path = "tests/reduce_sum_backward_smoke.rs"
[[test]]
name = "reduce_sum_dtype_smoke"
path = "tests/reduce_sum_dtype_smoke.rs"
[[test]]
name = "reduce_sum_smoke"
path = "tests/reduce_sum_smoke.rs"
[[test]]
name = "reduce_sum_to_smoke"
path = "tests/reduce_sum_to_smoke.rs"
[[test]]
name = "reduce_to_extras_smoke"
path = "tests/reduce_to_extras_smoke.rs"
[[test]]
name = "reduce_to_plan_smoke"
path = "tests/reduce_to_plan_smoke.rs"
[[test]]
name = "reduce_var_backward_smoke"
path = "tests/reduce_var_backward_smoke.rs"
[[test]]
name = "reduce_var_std_smoke"
path = "tests/reduce_var_std_smoke.rs"
[[test]]
name = "repeat_backward_smoke"
path = "tests/repeat_backward_smoke.rs"
[[test]]
name = "repeat_dtype_smoke"
path = "tests/repeat_dtype_smoke.rs"
[[test]]
name = "repeat_smoke"
path = "tests/repeat_smoke.rs"
[[test]]
name = "rfft_2d_smoke"
path = "tests/rfft_2d_smoke.rs"
[[test]]
name = "rfft_3d_smoke"
path = "tests/rfft_3d_smoke.rs"
[[test]]
name = "ring_attention_smoke"
path = "tests/ring_attention_smoke.rs"
[[test]]
name = "rms_norm_backward_smoke"
path = "tests/rms_norm_backward_smoke.rs"
[[test]]
name = "rms_norm_inplace_smoke"
path = "tests/rms_norm_inplace_smoke.rs"
[[test]]
name = "rms_norm_smoke"
path = "tests/rms_norm_smoke.rs"
[[test]]
name = "roi_align_smoke"
path = "tests/roi_align_smoke.rs"
[[test]]
name = "roi_pool_smoke"
path = "tests/roi_pool_smoke.rs"
[[test]]
name = "roll_backward_smoke"
path = "tests/roll_backward_smoke.rs"
[[test]]
name = "roll_dtype_smoke"
path = "tests/roll_dtype_smoke.rs"
[[test]]
name = "roll_smoke"
path = "tests/roll_smoke.rs"
[[test]]
name = "rope_apply_extras_smoke"
path = "tests/rope_apply_extras_smoke.rs"
[[test]]
name = "rope_apply_smoke"
path = "tests/rope_apply_smoke.rs"
[[test]]
name = "rope_backward_smoke"
path = "tests/rope_backward_smoke.rs"
[[test]]
name = "rope_scaling_smoke"
path = "tests/rope_scaling_smoke.rs"
[[test]]
name = "rope_smoke"
path = "tests/rope_smoke.rs"
[[test]]
name = "rope_strided_smoke"
path = "tests/rope_strided_smoke.rs"
[[test]]
name = "s4_rcr_smoke"
path = "tests/s4_rcr_smoke.rs"
[[test]]
name = "scan_cummax_backward_smoke"
path = "tests/scan_cummax_backward_smoke.rs"
[[test]]
name = "scan_cummax_smoke"
path = "tests/scan_cummax_smoke.rs"
[[test]]
name = "scan_cummin_backward_smoke"
path = "tests/scan_cummin_backward_smoke.rs"
[[test]]
name = "scan_cummin_smoke"
path = "tests/scan_cummin_smoke.rs"
[[test]]
name = "scan_cumprod_backward_smoke"
path = "tests/scan_cumprod_backward_smoke.rs"
[[test]]
name = "scan_cumprod_smoke"
path = "tests/scan_cumprod_smoke.rs"
[[test]]
name = "scan_cumsum_backward_smoke"
path = "tests/scan_cumsum_backward_smoke.rs"
[[test]]
name = "scan_cumsum_smoke"
path = "tests/scan_cumsum_smoke.rs"
[[test]]
name = "scan_log_cumsum_exp_backward_smoke"
path = "tests/scan_log_cumsum_exp_backward_smoke.rs"
[[test]]
name = "scan_log_cumsum_exp_smoke"
path = "tests/scan_log_cumsum_exp_smoke.rs"
[[test]]
name = "scatter_add_backward_smoke"
path = "tests/scatter_add_backward_smoke.rs"
[[test]]
name = "scatter_add_smoke"
path = "tests/scatter_add_smoke.rs"
[[test]]
name = "scatter_smoke"
path = "tests/scatter_smoke.rs"
[[test]]
name = "sdpa_backward_smoke"
path = "tests/sdpa_backward_smoke.rs"
[[test]]
name = "sdpa_block_sparse_smoke"
path = "tests/sdpa_block_sparse_smoke.rs"
[[test]]
name = "sdpa_bw_gqa_smoke"
path = "tests/sdpa_bw_gqa_smoke.rs"
[[test]]
name = "sdpa_smoke"
path = "tests/sdpa_smoke.rs"
[[test]]
name = "sdpa_strided_smoke"
path = "tests/sdpa_strided_smoke.rs"
[[test]]
name = "searchsorted_smoke"
path = "tests/searchsorted_smoke.rs"
[[test]]
name = "segment_max_backward_smoke"
path = "tests/segment_max_backward_smoke.rs"
[[test]]
name = "segment_max_smoke"
path = "tests/segment_max_smoke.rs"
[[test]]
name = "segment_mean_backward_smoke"
path = "tests/segment_mean_backward_smoke.rs"
[[test]]
name = "segment_mean_smoke"
path = "tests/segment_mean_smoke.rs"
[[test]]
name = "segment_min_backward_smoke"
path = "tests/segment_min_backward_smoke.rs"
[[test]]
name = "segment_min_smoke"
path = "tests/segment_min_smoke.rs"
[[test]]
name = "segment_prod_backward_smoke"
path = "tests/segment_prod_backward_smoke.rs"
[[test]]
name = "segment_prod_smoke"
path = "tests/segment_prod_smoke.rs"
[[test]]
name = "segment_sum_backward_smoke"
path = "tests/segment_sum_backward_smoke.rs"
[[test]]
name = "segment_sum_smoke"
path = "tests/segment_sum_smoke.rs"
[[test]]
name = "selective_scan_bw_smoke"
path = "tests/selective_scan_bw_smoke.rs"
[[test]]
name = "selective_scan_smoke"
path = "tests/selective_scan_smoke.rs"
[[test]]
name = "smoothquant_smoke"
path = "tests/smoothquant_smoke.rs"
[[test]]
name = "softmax_backward_smoke"
path = "tests/softmax_backward_smoke.rs"
[[test]]
name = "softmax_inplace_smoke"
path = "tests/softmax_inplace_smoke.rs"
[[test]]
name = "softmax_smoke"
path = "tests/softmax_smoke.rs"
[[test]]
name = "sort_backward_smoke"
path = "tests/sort_backward_smoke.rs"
[[test]]
name = "sort_smoke"
path = "tests/sort_smoke.rs"
[[test]]
name = "sparse24_throughput_smoke"
path = "tests/sparse24_throughput_smoke.rs"
[[test]]
name = "sparsemax_backward_smoke"
path = "tests/sparsemax_backward_smoke.rs"
[[test]]
name = "sparsemax_smoke"
path = "tests/sparsemax_smoke.rs"
[[test]]
name = "ssd_chunk_scan_bw_smoke"
path = "tests/ssd_chunk_scan_bw_smoke.rs"
[[test]]
name = "ssd_chunk_scan_smoke"
path = "tests/ssd_chunk_scan_smoke.rs"
[[test]]
name = "strided_siblings_ffi_smoke"
path = "tests/strided_siblings_ffi_smoke.rs"
[[test]]
name = "ternary_addcdiv_backward_smoke"
path = "tests/ternary_addcdiv_backward_smoke.rs"
[[test]]
name = "ternary_addcdiv_smoke"
path = "tests/ternary_addcdiv_smoke.rs"
[[test]]
name = "ternary_addcmul_backward_smoke"
path = "tests/ternary_addcmul_backward_smoke.rs"
[[test]]
name = "ternary_addcmul_smoke"
path = "tests/ternary_addcmul_smoke.rs"
[[test]]
name = "ternary_clamp_backward_smoke"
path = "tests/ternary_clamp_backward_smoke.rs"
[[test]]
name = "ternary_clamp_dtype_smoke"
path = "tests/ternary_clamp_dtype_smoke.rs"
[[test]]
name = "ternary_clamp_smoke"
path = "tests/ternary_clamp_smoke.rs"
[[test]]
name = "ternary_fma_backward_smoke"
path = "tests/ternary_fma_backward_smoke.rs"
[[test]]
name = "ternary_fma_smoke"
path = "tests/ternary_fma_smoke.rs"
[[test]]
name = "topk_sampling_smoke"
path = "tests/topk_sampling_smoke.rs"
[[test]]
name = "topk_smoke"
path = "tests/topk_smoke.rs"
[[test]]
name = "trace_smoke"
path = "tests/trace_smoke.rs"
[[test]]
name = "triu_tril_smoke"
path = "tests/triu_tril_smoke.rs"
[[test]]
name = "triu_tril_strided_smoke"
path = "tests/triu_tril_strided_smoke.rs"
[[test]]
name = "u4_rcr_smoke"
path = "tests/u4_rcr_smoke.rs"
[[test]]
name = "unary_abs_smoke"
path = "tests/unary_abs_smoke.rs"
[[test]]
name = "unary_acos_backward_smoke"
path = "tests/unary_acos_backward_smoke.rs"
[[test]]
name = "unary_acos_smoke"
path = "tests/unary_acos_smoke.rs"
[[test]]
name = "unary_acosh_backward_smoke"
path = "tests/unary_acosh_backward_smoke.rs"
[[test]]
name = "unary_acosh_smoke"
path = "tests/unary_acosh_smoke.rs"
[[test]]
name = "unary_asin_backward_smoke"
path = "tests/unary_asin_backward_smoke.rs"
[[test]]
name = "unary_asin_smoke"
path = "tests/unary_asin_smoke.rs"
[[test]]
name = "unary_asinh_backward_smoke"
path = "tests/unary_asinh_backward_smoke.rs"
[[test]]
name = "unary_asinh_smoke"
path = "tests/unary_asinh_smoke.rs"
[[test]]
name = "unary_atan_backward_smoke"
path = "tests/unary_atan_backward_smoke.rs"
[[test]]
name = "unary_atan_smoke"
path = "tests/unary_atan_smoke.rs"
[[test]]
name = "unary_atanh_backward_smoke"
path = "tests/unary_atanh_backward_smoke.rs"
[[test]]
name = "unary_atanh_smoke"
path = "tests/unary_atanh_smoke.rs"
[[test]]
name = "unary_cbrt_smoke"
path = "tests/unary_cbrt_smoke.rs"
[[test]]
name = "unary_ceil_smoke"
path = "tests/unary_ceil_smoke.rs"
[[test]]
name = "unary_cos_backward_smoke"
path = "tests/unary_cos_backward_smoke.rs"
[[test]]
name = "unary_cos_smoke"
path = "tests/unary_cos_smoke.rs"
[[test]]
name = "unary_cosh_backward_smoke"
path = "tests/unary_cosh_backward_smoke.rs"
[[test]]
name = "unary_cosh_smoke"
path = "tests/unary_cosh_smoke.rs"
[[test]]
name = "unary_cube_backward_smoke"
path = "tests/unary_cube_backward_smoke.rs"
[[test]]
name = "unary_cube_smoke"
path = "tests/unary_cube_smoke.rs"
[[test]]
name = "unary_elu_alpha_smoke"
path = "tests/unary_elu_alpha_smoke.rs"
[[test]]
name = "unary_elu_backward_smoke"
path = "tests/unary_elu_backward_smoke.rs"
[[test]]
name = "unary_elu_smoke"
path = "tests/unary_elu_smoke.rs"
[[test]]
name = "unary_erf_backward_smoke"
path = "tests/unary_erf_backward_smoke.rs"
[[test]]
name = "unary_erf_smoke"
path = "tests/unary_erf_smoke.rs"
[[test]]
name = "unary_erfc_backward_smoke"
path = "tests/unary_erfc_backward_smoke.rs"
[[test]]
name = "unary_erfc_smoke"
path = "tests/unary_erfc_smoke.rs"
[[test]]
name = "unary_exp2_backward_smoke"
path = "tests/unary_exp2_backward_smoke.rs"
[[test]]
name = "unary_exp2_smoke"
path = "tests/unary_exp2_smoke.rs"
[[test]]
name = "unary_exp_backward_smoke"
path = "tests/unary_exp_backward_smoke.rs"
[[test]]
name = "unary_exp_smoke"
path = "tests/unary_exp_smoke.rs"
[[test]]
name = "unary_expm1_backward_smoke"
path = "tests/unary_expm1_backward_smoke.rs"
[[test]]
name = "unary_expm1_smoke"
path = "tests/unary_expm1_smoke.rs"
[[test]]
name = "unary_floor_smoke"
path = "tests/unary_floor_smoke.rs"
[[test]]
name = "unary_frac_smoke"
path = "tests/unary_frac_smoke.rs"
[[test]]
name = "unary_gelu_backward_smoke"
path = "tests/unary_gelu_backward_smoke.rs"
[[test]]
name = "unary_gelu_erf_smoke"
path = "tests/unary_gelu_erf_smoke.rs"
[[test]]
name = "unary_gelu_smoke"
path = "tests/unary_gelu_smoke.rs"
[[test]]
name = "unary_gelu_tanh_backward_smoke"
path = "tests/unary_gelu_tanh_backward_smoke.rs"
[[test]]
name = "unary_gelu_tanh_smoke"
path = "tests/unary_gelu_tanh_smoke.rs"
[[test]]
name = "unary_hardshrink_backward_smoke"
path = "tests/unary_hardshrink_backward_smoke.rs"
[[test]]
name = "unary_hardshrink_smoke"
path = "tests/unary_hardshrink_smoke.rs"
[[test]]
name = "unary_hardsigmoid_backward_smoke"
path = "tests/unary_hardsigmoid_backward_smoke.rs"
[[test]]
name = "unary_hardsigmoid_smoke"
path = "tests/unary_hardsigmoid_smoke.rs"
[[test]]
name = "unary_hardswish_backward_smoke"
path = "tests/unary_hardswish_backward_smoke.rs"
[[test]]
name = "unary_hardswish_smoke"
path = "tests/unary_hardswish_smoke.rs"
[[test]]
name = "unary_hardtanh_backward_smoke"
path = "tests/unary_hardtanh_backward_smoke.rs"
[[test]]
name = "unary_hardtanh_smoke"
path = "tests/unary_hardtanh_smoke.rs"
[[test]]
name = "unary_leaky_relu_backward_smoke"
path = "tests/unary_leaky_relu_backward_smoke.rs"
[[test]]
name = "unary_leaky_relu_smoke"
path = "tests/unary_leaky_relu_smoke.rs"
[[test]]
name = "unary_lgamma_smoke"
path = "tests/unary_lgamma_smoke.rs"
[[test]]
name = "unary_log10_backward_smoke"
path = "tests/unary_log10_backward_smoke.rs"
[[test]]
name = "unary_log10_smoke"
path = "tests/unary_log10_smoke.rs"
[[test]]
name = "unary_log1p_backward_smoke"
path = "tests/unary_log1p_backward_smoke.rs"
[[test]]
name = "unary_log1p_smoke"
path = "tests/unary_log1p_smoke.rs"
[[test]]
name = "unary_log2_backward_smoke"
path = "tests/unary_log2_backward_smoke.rs"
[[test]]
name = "unary_log2_smoke"
path = "tests/unary_log2_smoke.rs"
[[test]]
name = "unary_log_backward_smoke"
path = "tests/unary_log_backward_smoke.rs"
[[test]]
name = "unary_log_smoke"
path = "tests/unary_log_smoke.rs"
[[test]]
name = "unary_logit_backward_smoke"
path = "tests/unary_logit_backward_smoke.rs"
[[test]]
name = "unary_logit_smoke"
path = "tests/unary_logit_smoke.rs"
[[test]]
name = "unary_mish_backward_smoke"
path = "tests/unary_mish_backward_smoke.rs"
[[test]]
name = "unary_mish_smoke"
path = "tests/unary_mish_smoke.rs"
[[test]]
name = "unary_neg_dtype_smoke"
path = "tests/unary_neg_dtype_smoke.rs"
[[test]]
name = "unary_neg_smoke"
path = "tests/unary_neg_smoke.rs"
[[test]]
name = "unary_powf_smoke"
path = "tests/unary_powf_smoke.rs"
[[test]]
name = "unary_powi_smoke"
path = "tests/unary_powi_smoke.rs"
[[test]]
name = "unary_reciprocal_backward_smoke"
path = "tests/unary_reciprocal_backward_smoke.rs"
[[test]]
name = "unary_reciprocal_smoke"
path = "tests/unary_reciprocal_smoke.rs"
[[test]]
name = "unary_relu6_backward_smoke"
path = "tests/unary_relu6_backward_smoke.rs"
[[test]]
name = "unary_relu6_smoke"
path = "tests/unary_relu6_smoke.rs"
[[test]]
name = "unary_relu_backward_smoke"
path = "tests/unary_relu_backward_smoke.rs"
[[test]]
name = "unary_relu_smoke"
path = "tests/unary_relu_smoke.rs"
[[test]]
name = "unary_round_smoke"
path = "tests/unary_round_smoke.rs"
[[test]]
name = "unary_rsqrt_backward_smoke"
path = "tests/unary_rsqrt_backward_smoke.rs"
[[test]]
name = "unary_rsqrt_smoke"
path = "tests/unary_rsqrt_smoke.rs"
[[test]]
name = "unary_selu_backward_smoke"
path = "tests/unary_selu_backward_smoke.rs"
[[test]]
name = "unary_selu_smoke"
path = "tests/unary_selu_smoke.rs"
[[test]]
name = "unary_sigmoid_backward_smoke"
path = "tests/unary_sigmoid_backward_smoke.rs"
[[test]]
name = "unary_sigmoid_smoke"
path = "tests/unary_sigmoid_smoke.rs"
[[test]]
name = "unary_sign_smoke"
path = "tests/unary_sign_smoke.rs"
[[test]]
name = "unary_silu_backward_smoke"
path = "tests/unary_silu_backward_smoke.rs"
[[test]]
name = "unary_silu_smoke"
path = "tests/unary_silu_smoke.rs"
[[test]]
name = "unary_sin_backward_smoke"
path = "tests/unary_sin_backward_smoke.rs"
[[test]]
name = "unary_sin_smoke"
path = "tests/unary_sin_smoke.rs"
[[test]]
name = "unary_sinh_backward_smoke"
path = "tests/unary_sinh_backward_smoke.rs"
[[test]]
name = "unary_sinh_smoke"
path = "tests/unary_sinh_smoke.rs"
[[test]]
name = "unary_softplus_backward_smoke"
path = "tests/unary_softplus_backward_smoke.rs"
[[test]]
name = "unary_softplus_smoke"
path = "tests/unary_softplus_smoke.rs"
[[test]]
name = "unary_softshrink_backward_smoke"
path = "tests/unary_softshrink_backward_smoke.rs"
[[test]]
name = "unary_softshrink_smoke"
path = "tests/unary_softshrink_smoke.rs"
[[test]]
name = "unary_softsign_smoke"
path = "tests/unary_softsign_smoke.rs"
[[test]]
name = "unary_sqrt_backward_smoke"
path = "tests/unary_sqrt_backward_smoke.rs"
[[test]]
name = "unary_sqrt_smoke"
path = "tests/unary_sqrt_smoke.rs"
[[test]]
name = "unary_square_backward_smoke"
path = "tests/unary_square_backward_smoke.rs"
[[test]]
name = "unary_square_smoke"
path = "tests/unary_square_smoke.rs"
[[test]]
name = "unary_step_smoke"
path = "tests/unary_step_smoke.rs"
[[test]]
name = "unary_tan_backward_smoke"
path = "tests/unary_tan_backward_smoke.rs"
[[test]]
name = "unary_tan_smoke"
path = "tests/unary_tan_smoke.rs"
[[test]]
name = "unary_tanh_backward_smoke"
path = "tests/unary_tanh_backward_smoke.rs"
[[test]]
name = "unary_tanh_smoke"
path = "tests/unary_tanh_smoke.rs"
[[test]]
name = "unary_tanhshrink_backward_smoke"
path = "tests/unary_tanhshrink_backward_smoke.rs"
[[test]]
name = "unary_tanhshrink_smoke"
path = "tests/unary_tanhshrink_smoke.rs"
[[test]]
name = "unary_threshold_backward_smoke"
path = "tests/unary_threshold_backward_smoke.rs"
[[test]]
name = "unary_threshold_smoke"
path = "tests/unary_threshold_smoke.rs"
[[test]]
name = "unary_trunc_smoke"
path = "tests/unary_trunc_smoke.rs"
[[test]]
name = "unique_consecutive_smoke"
path = "tests/unique_consecutive_smoke.rs"
[[test]]
name = "unsorted_segment_max_backward_smoke"
path = "tests/unsorted_segment_max_backward_smoke.rs"
[[test]]
name = "unsorted_segment_max_smoke"
path = "tests/unsorted_segment_max_smoke.rs"
[[test]]
name = "unsorted_segment_mean_backward_smoke"
path = "tests/unsorted_segment_mean_backward_smoke.rs"
[[test]]
name = "unsorted_segment_mean_smoke"
path = "tests/unsorted_segment_mean_smoke.rs"
[[test]]
name = "unsorted_segment_min_backward_smoke"
path = "tests/unsorted_segment_min_backward_smoke.rs"
[[test]]
name = "unsorted_segment_min_smoke"
path = "tests/unsorted_segment_min_smoke.rs"
[[test]]
name = "unsorted_segment_prod_backward_smoke"
path = "tests/unsorted_segment_prod_backward_smoke.rs"
[[test]]
name = "unsorted_segment_prod_smoke"
path = "tests/unsorted_segment_prod_smoke.rs"
[[test]]
name = "unsorted_segment_sum_backward_smoke"
path = "tests/unsorted_segment_sum_backward_smoke.rs"
[[test]]
name = "unsorted_segment_sum_smoke"
path = "tests/unsorted_segment_sum_smoke.rs"
[[test]]
name = "upsample_ffi_smoke"
path = "tests/upsample_ffi_smoke.rs"
[[test]]
name = "where_backward_smoke"
path = "tests/where_backward_smoke.rs"
[[test]]
name = "where_cond_dtype_fanout_smoke"
path = "tests/where_cond_dtype_fanout_smoke.rs"
[[test]]
name = "where_dtype_smoke"
path = "tests/where_dtype_smoke.rs"
[[test]]
name = "where_smoke"
path = "tests/where_smoke.rs"
[[test]]
name = "write_slice_smoke"
path = "tests/write_slice_smoke.rs"
[dependencies.baracuda-cutlass]
version = "0.0.1-alpha.68"
[dependencies.baracuda-driver]
version = "0.0.1-alpha.68"
[dependencies.baracuda-kernels-sys]
version = "0.0.1-alpha.68"
[dependencies.baracuda-kernels-types]
version = "0.0.1-alpha.68"
[dependencies.baracuda-megatron]
version = "0.0.1-alpha.68"
optional = true
[dependencies.baracuda-nccl]
version = "0.0.1-alpha.68"
optional = true
[dependencies.baracuda-nccl-sys]
version = "0.0.1-alpha.68"
optional = true
[dependencies.baracuda-nvshmem]
version = "0.0.1-alpha.68"
optional = true
[dependencies.baracuda-nvshmem-sys]
version = "0.0.1-alpha.68"
optional = true
[dependencies.baracuda-optim]
version = "0.0.1-alpha.68"
optional = true
[dependencies.baracuda-transformer-engine]
version = "0.0.1-alpha.68"
optional = true
[dependencies.baracuda-types]
version = "0.0.1-alpha.68"
features = [
"half-crate",
"f8-crate",
]
[dependencies.half]
version = "2"
default-features = false
[dependencies.thiserror]
version = "1"
[dev-dependencies.float8]
version = "0.7"
default-features = false
[dev-dependencies.libm]
version = "0.2"
[lints.clippy]
too_many_arguments = "allow"
type_complexity = "allow"
[lints.clippy.all]
level = "warn"
priority = -1
[lints.rust]
missing_debug_implementations = "warn"
missing_docs = "deny"
unreachable_pub = "warn"