InfiniOps
Operator Library for Accelerators
Loading...
Searching...
No Matches
base Directory Reference

Files

 abs.h
 
 absolute.h
 
 acos.h
 
 acosh.h
 
 adaptive_avg_pool2d.h
 
 adaptive_avg_pool3d.h
 
 adaptive_avg_pool3d_backward.h
 
 adaptive_max_pool2d.h
 
 adaptive_max_pool2d_backward.h
 
 adaptive_max_pool3d.h
 
 adaptive_max_pool3d_backward.h
 
 add.h
 
 add_rms_norm.h
 
 addbmm.h
 
 addcdiv.h
 
 addcmul.h
 
 addmm.h
 
 addmv.h
 
 addr.h
 
 amax.h
 
 amin.h
 
 aminmax.h
 
 angle.h
 
 arccos.h
 
 arccosh.h
 
 arcsin.h
 
 arcsinh.h
 
 arctan.h
 
 arctan2.h
 
 arctanh.h
 
 argmax.h
 
 argmin.h
 
 argsort.h
 
 asin.h
 
 asinh.h
 
 atan.h
 
 atan2.h
 
 atanh.h
 
 avg_pool2d.h
 
 avg_pool2d_backward.h
 
 avg_pool3d.h
 
 avg_pool3d_backward.h
 
 awq_marlin_repack.h
 
 baddbmm.h
 
 batch_norm_elemt.h
 
 bernoulli.h
 
 binary_cross_entropy.h
 
 binary_cross_entropy_backward.h
 
 bitwise_and.h
 
 bitwise_left_shift.h
 
 bitwise_not.h
 
 bitwise_or.h
 
 bitwise_right_shift.h
 
 bitwise_xor.h
 
 bmm.h
 
 bucketize.h
 
 cast.h
 
 cat.h
 
 causal_softmax.h
 
 causal_softmax_infinilm.h
 
 ceil.h
 
 chain_matmul.h
 
 cholesky.h
 
 cholesky_inverse.h
 
 cholesky_solve.h
 
 clamp.h
 
 clamp_max.h
 
 clamp_min.h
 
 clip.h
 
 col2im.h
 
 column_stack.h
 
 complex.h
 
 concatenate.h
 
 conj_physical.h
 
 conv1d.h
 
 conv2d.h
 
 conv3d.h
 
 conv_infinilm.h
 
 convolution.h
 
 copy.h
 
 copysign.h
 
 cos.h
 
 cosh.h
 
 cross.h
 
 cudnn_convolution.h
 
 cummax.h
 
 cummin.h
 
 cumprod.h
 
 cumsum.h
 
 cutlass_scaled_mm.h
 
 deg2rad.h
 
 diag.h
 
 diff.h
 
 digamma.h
 
 div.h
 
 divide.h
 
 dot.h
 
 dstack.h
 
 elu.h
 
 elu_backward.h
 
 embedding.h
 
 eq.h
 
 erf.h
 
 erfc.h
 
 erfinv.h
 
 exp.h
 
 exp2.h
 
 expm1.h
 
 fft_fft.h
 
 fft_fft2.h
 
 fft_fftn.h
 
 fft_hfft.h
 
 fft_hfft2.h
 
 fft_hfftn.h
 
 fft_ifft.h
 
 fft_ifft2.h
 
 fft_ifftn.h
 
 fft_ihfft.h
 
 fft_ihfft2.h
 
 fft_ihfftn.h
 
 fft_irfft.h
 
 fft_irfft2.h
 
 fft_irfftn.h
 
 fft_rfft.h
 
 fft_rfft2.h
 
 fft_rfftn.h
 
 fill.h
 
 fix.h
 
 flash_attention.h
 
 flash_attn_varlen_func.h
 
 flash_attn_with_kvcache.h
 
 float_power.h
 
 floor.h
 
 floor_divide.h
 
 fmax.h
 
 fmin.h
 
 fmod.h
 
 frac.h
 
 fractional_max_pool2d.h
 
 fractional_max_pool2d_backward.h
 
 fractional_max_pool3d.h
 
 fractional_max_pool3d_backward.h
 
 frexp.h
 
 frobenius_norm.h
 
 fused_add_rms_norm.h
 
 gather.h
 
 gcd.h
 
 ge.h
 
 gelu.h
 
 gelu_backward.h
 
 gelu_infinilm.h
 
 gelutanh_infinilm.h
 
 gemm.h
 
 geqrf.h
 
 ger.h
 
 get_cutlass_moe_mm_data.h
 
 glu.h
 
 glu_backward.h
 
 gptq_marlin_repack.h
 
 greater.h
 
 greater_equal.h
 
 grouped_topk.h
 
 gt.h
 
 hardshrink.h
 
 hardshrink_backward.h
 
 hardsigmoid.h
 
 hardsigmoid_backward.h
 
 hardswish.h
 
 hardtanh.h
 
 hardtanh_backward.h
 
 heaviside.h
 
 histc.h
 
 histogram.h
 
 hspmm.h
 
 hstack.h
 
 huber_loss.h
 
 huber_loss_backward.h
 
 hypot.h
 
 i0.h
 
 igamma.h
 
 igammac.h
 
 im2col.h
 
 index.h
 
 index_add.h
 
 index_copy.h
 
 index_fill.h
 
 index_reduce.h
 
 index_select.h
 
 inner.h
 
 internal_add_relu.h
 
 internal_addmm_activation.h
 
 internal_batch_norm_with_update.h
 
 internal_chunk_cat.h
 
 internal_compute_linear_combination.h
 
 internal_conv_depthwise2d.h
 
 internal_convert_indices_from_coo_to_csr.h
 
 internal_convert_indices_from_csr_to_coo.h
 
 internal_fft_c2c.h
 
 internal_fft_c2r.h
 
 internal_fft_r2c.h
 
 internal_int_mm.h
 
 internal_linalg_det.h
 
 internal_linalg_eigh.h
 
 internal_linalg_slogdet.h
 
 internal_linalg_solve_ex.h
 
 internal_linalg_svd.h
 
 internal_log_softmax.h
 
 internal_log_softmax_backward_data.h
 
 internal_logcumsumexp.h
 
 internal_native_batch_norm_legit.h
 
 internal_slow_conv2d_backward.h
 
 internal_slow_conv2d_forward.h
 
 internal_softmax.h
 
 internal_softmax_backward_data.h
 
 internal_stack.h
 
 internal_upsample_bicubic2d_aa.h
 
 internal_upsample_bicubic2d_aa_backward.h
 
 internal_upsample_bilinear2d_aa.h
 
 internal_upsample_bilinear2d_aa_backward.h
 
 internal_upsample_nearest_exact1d.h
 
 internal_upsample_nearest_exact1d_backward.h
 
 internal_upsample_nearest_exact2d.h
 
 internal_upsample_nearest_exact2d_backward.h
 
 internal_upsample_nearest_exact3d.h
 
 internal_upsample_nearest_exact3d_backward.h
 
 inverse.h
 
 isin.h
 
 isneginf.h
 
 isposinf.h
 
 kron.h
 
 kthvalue.h
 
 kv_caching_infinilm.h
 
 lcm.h
 
 ldexp.h
 
 le.h
 
 leaky_relu.h
 
 leaky_relu_backward.h
 
 lerp.h
 
 less.h
 
 less_equal.h
 
 lgamma.h
 
 linalg_cholesky.h
 
 linalg_cholesky_ex.h
 
 linalg_cond.h
 
 linalg_cross.h
 
 linalg_det.h
 
 linalg_eig.h
 
 linalg_eigh.h
 
 linalg_eigvals.h
 
 linalg_eigvalsh.h
 
 linalg_householder_product.h
 
 linalg_inv.h
 
 linalg_inv_ex.h
 
 linalg_ldl_factor.h
 
 linalg_ldl_factor_ex.h
 
 linalg_ldl_solve.h
 
 linalg_lstsq.h
 
 linalg_lu.h
 
 linalg_lu_factor.h
 
 linalg_lu_factor_ex.h
 
 linalg_lu_solve.h
 
 linalg_matmul.h
 
 linalg_matrix_norm.h
 
 linalg_matrix_power.h
 
 linalg_matrix_rank.h
 
 linalg_multi_dot.h
 
 linalg_norm.h
 
 linalg_pinv.h
 
 linalg_qr.h
 
 linalg_slogdet.h
 
 linalg_solve.h
 
 linalg_solve_ex.h
 
 linalg_solve_triangular.h
 
 linalg_svd.h
 
 linalg_svdvals.h
 
 linalg_tensorinv.h
 
 linalg_tensorsolve.h
 
 linalg_vecdot.h
 
 linalg_vector_norm.h
 
 linear.h
 
 linspace.h
 
 log.h
 
 log10.h
 
 log1p.h
 
 log2.h
 
 log_sigmoid.h
 
 log_sigmoid_backward.h
 
 log_sigmoid_forward.h
 
 log_softmax.h
 
 logaddexp.h
 
 logaddexp2.h
 
 logcumsumexp.h
 
 logical_and.h
 
 logical_not.h
 
 logical_or.h
 
 logical_xor.h
 
 logit.h
 
 logit_backward.h
 
 logspace.h
 
 logsumexp.h
 
 lt.h
 
 lu_solve.h
 
 lu_unpack.h
 
 masked_fill.h
 
 masked_select.h
 
 matmul.h
 
 matrix_power.h
 
 max.h
 
 max_pool2d_with_indices.h
 
 max_pool2d_with_indices_backward.h
 
 max_pool3d_with_indices.h
 
 max_pool3d_with_indices_backward.h
 
 max_unpool2d.h
 
 max_unpool3d.h
 
 maximum.h
 
 mean.h
 
 median.h
 
 min.h
 
 minimum.h
 
 mish.h
 
 mkldnn_adaptive_avg_pool2d.h
 
 mm.h
 
 mode.h
 
 moe_align_block_size.h
 
 moe_sum.h
 
 moe_wna16_gemm.h
 
 moe_wna16_marlin_gemm.h
 
 mse_loss.h
 
 mse_loss_backward.h
 
 msort.h
 
 mul.h
 
 multi_margin_loss.h
 
 multi_margin_loss_backward.h
 
 multilabel_margin_loss.h
 
 multilabel_margin_loss_backward.h
 
 multilabel_margin_loss_forward.h
 
 multinomial.h
 
 multiply.h
 
 mv.h
 
 mvlgamma.h
 
 nan_to_num.h
 
 nanmean.h
 
 nanmedian.h
 
 nanquantile.h
 
 nansum.h
 
 narrow_copy.h
 
 native_batch_norm.h
 
 ne.h
 
 neg.h
 
 negative.h
 
 nextafter.h
 
 nll_loss.h
 
 nll_loss2d.h
 
 nll_loss2d_backward.h
 
 nll_loss2d_forward.h
 
 nll_loss_backward.h
 
 nll_loss_forward.h
 
 nonzero.h
 
 nonzero_static.h
 
 norm.h
 
 normal.h
 
 not_equal.h
 
 nuclear_norm.h
 
 orgqr.h
 
 ormqr.h
 
 outer.h
 
 paged_attention_infinilm.h
 
 paged_attention_prefill_infinilm.h
 
 paged_caching_infinilm.h
 
 polar.h
 
 pow.h
 
 prod.h
 
 qr.h
 
 quantile.h
 
 rad2deg.h
 
 random.h
 
 random_sample_infinilm.h
 
 rearrange_infinilm.h
 
 reciprocal.h
 
 reflection_pad1d.h
 
 reflection_pad1d_backward.h
 
 reflection_pad2d.h
 
 reflection_pad2d_backward.h
 
 reflection_pad3d.h
 
 reflection_pad3d_backward.h
 
 relu.h
 
 relu_infinilm.h
 
 remainder.h
 
 renorm.h
 
 replication_pad1d.h
 
 replication_pad1d_backward.h
 
 replication_pad2d.h
 
 replication_pad2d_backward.h
 
 replication_pad3d.h
 
 replication_pad3d_backward.h
 
 reshape_and_cache.h
 
 reshape_and_cache_flash.h
 
 rms_norm.h
 
 rotary_embedding.h
 
 rotary_embedding_infinilm.h
 
 round.h
 
 row_stack.h
 
 rrelu_with_noise.h
 
 rsqrt.h
 
 scaled_dot_product_attention.h
 
 scaled_softmax.h
 
 scaled_softmax_infinilm.h
 
 scatter.h
 
 scatter_add.h
 
 scatter_reduce.h
 
 searchsorted.h
 
 set.h
 
 sgn.h
 
 sigmoid.h
 
 sigmoid_backward.h
 
 sigmoid_infinilm.h
 
 sign.h
 
 signbit.h
 
 silu.h
 
 silu_and_mul.h
 
 silu_and_mul_infinilm.h
 
 silu_backward.h
 
 sin.h
 
 sinc.h
 
 sinh.h
 
 slogdet.h
 
 slow_conv3d.h
 
 slow_conv3d_forward.h
 
 slow_conv_transpose2d.h
 
 slow_conv_transpose3d.h
 
 smooth_l1_loss.h
 
 smooth_l1_loss_backward.h
 
 soft_margin_loss.h
 
 soft_margin_loss_backward.h
 
 softmax.h
 
 softmax_infinilm.h
 
 softplus.h
 
 softplus_backward.h
 
 softshrink.h
 
 softshrink_backward.h
 
 sort.h
 
 sparse_sampled_addmm.h
 
 special_airy_ai.h
 
 special_bessel_j0.h
 
 special_bessel_j1.h
 
 special_bessel_y0.h
 
 special_bessel_y1.h
 
 special_chebyshev_polynomial_t.h
 
 special_chebyshev_polynomial_u.h
 
 special_chebyshev_polynomial_v.h
 
 special_chebyshev_polynomial_w.h
 
 special_digamma.h
 
 special_entr.h
 
 special_erf.h
 
 special_erfc.h
 
 special_erfcx.h
 
 special_erfinv.h
 
 special_exp2.h
 
 special_expit.h
 
 special_expm1.h
 
 special_gammainc.h
 
 special_gammaincc.h
 
 special_gammaln.h
 
 special_hermite_polynomial_h.h
 
 special_hermite_polynomial_he.h
 
 special_i0.h
 
 special_i0e.h
 
 special_i1.h
 
 special_i1e.h
 
 special_laguerre_polynomial_l.h
 
 special_legendre_polynomial_p.h
 
 special_log1p.h
 
 special_log_ndtr.h
 
 special_logit.h
 
 special_logsumexp.h
 
 special_modified_bessel_i0.h
 
 special_modified_bessel_i1.h
 
 special_modified_bessel_k0.h
 
 special_modified_bessel_k1.h
 
 special_multigammaln.h
 
 special_ndtr.h
 
 special_ndtri.h
 
 special_psi.h
 
 special_round.h
 
 special_scaled_modified_bessel_k0.h
 
 special_scaled_modified_bessel_k1.h
 
 special_shifted_chebyshev_polynomial_t.h
 
 special_shifted_chebyshev_polynomial_u.h
 
 special_shifted_chebyshev_polynomial_v.h
 
 special_shifted_chebyshev_polynomial_w.h
 
 special_sinc.h
 
 special_spherical_bessel_j0.h
 
 special_xlog1py.h
 
 special_xlogy.h
 
 special_zeta.h
 
 sqrt.h
 
 square.h
 
 sspaddmm.h
 
 stack.h
 
 std.h
 
 sub.h
 
 subtract.h
 
 sum.h
 
 svd.h
 
 swiglu.h
 
 take.h
 
 take_along_dim.h
 
 tan.h
 
 tanh.h
 
 tanh_backward.h
 
 tensordot.h
 
 thnn_conv2d.h
 
 threshold.h
 
 threshold_backward.h
 
 top_k_top_p_sample_infinilm.h
 
 top_k_top_p_sampler.h
 
 top_k_top_p_sampling_from_logits.h
 
 topk.h
 
 topk_sigmoid.h
 
 topk_softmax.h
 
 topksoftmax_infinilm.h
 
 triangular_solve.h
 
 tril.h
 
 triu.h
 
 true_divide.h
 
 trunc.h
 
 upsample_bicubic2d.h
 
 upsample_bicubic2d_backward.h
 
 upsample_bilinear2d.h
 
 upsample_bilinear2d_backward.h
 
 upsample_linear1d.h
 
 upsample_linear1d_backward.h
 
 upsample_nearest1d.h
 
 upsample_nearest1d_backward.h
 
 upsample_nearest2d.h
 
 upsample_nearest2d_backward.h
 
 upsample_nearest3d.h
 
 upsample_nearest3d_backward.h
 
 upsample_trilinear3d.h
 
 upsample_trilinear3d_backward.h
 
 var.h
 
 vdot.h
 
 vstack.h
 
 where.h
 
 xlogy.h
 
 zeros_infinilm.h