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
src
base
Generated by
1.9.8