Op-db Coverage per Op and Type

This page shows, for every op collected from torch.testing._internal.common_methods_invocations (op_db), which data types are covered by the op-db export tests in:

• unittests.torch.coverage.test_onnx_export_common_methods

• unittests.torch.coverage.test_onnx_export_common_methods_tracing

• unittests.torch.coverage.test_onnx_export_common_methods_new_tracing

Legend:

• ✔ — converter exists and the test passes for that dtype.

• ⚠ xfail — converter exists but the test is a known failure for that dtype (incorrect numerical results, unsupported dtype mapping, etc.).

• ✘ no converter — no ONNX converter has been implemented for this op yet.

• — — the op does not support that dtype at all.

The exclusion sets (NO_CONVERTER_OPS, XFAIL_OPS, and the per-dtype xfail sets) are defined in yobx.torch.coverage.op_coverage and are imported by the test module so that both sources always stay in sync.

Default export path

Default export path

Op	float32	float16	bfloat16	int32	int64
H	✘ no converter	✘ no converter	✘ no converter	✘ no converter	✘ no converter
T	✔	✔	✔	✔	✔
__getitem__	✔	✔	✔	✔	✔
__radd__	✔	✔	✔	✔	✔
__rand__	—	—	—	✔	✔
__rdiv__	✔	✔	✔	⚠ xfail	⚠ xfail
__rmatmul__	✔	✔	✔	✔	✔
__rmod__	✔	✔	✔	—	—
__rmul__	✔	✔	✔	✔	✔
__ror__	—	—	—	✔	✔
__rpow__	✔	✔	✔	⚠ xfail	⚠ xfail
__rsub__	✘ no converter	✘ no converter	✘ no converter	✘ no converter	✘ no converter
__rxor__	—	—	—	⚠ xfail	⚠ xfail
_batch_norm_with_update	✔	✔	✔	—	—
_chunk_cat	✔	✔	✔	✔	✔
_native_batch_norm_legit	✔	✔	✔	—	—
_softmax_backward_data	✔	✔	✔	—	—
_unsafe_masked_index	✔	✔	✔	✔	✔
_unsafe_masked_index_put_accumulate	✔	✔	✔	✔	✔
_upsample_bilinear2d_aa	✔	—	—	—	—
abs	✔	✔	✔	✔	✔
acos	✔	✔	✔	⚠ xfail	⚠ xfail
acosh	✔	✔	✔	⚠ xfail	⚠ xfail
add	✔	✔	✔	✔	✔
addbmm	✔	✔	✔	✔	✔
addcdiv	✘ no converter	✘ no converter	✘ no converter	—	—
addcmul	✔	⚠ xfail	⚠ xfail	✔	✔
addmm	✔	✔	✔	✔	✔
addmv	✔	✔	✔	✔	✔
addr	✘ no converter	✘ no converter	✘ no converter	✘ no converter	✘ no converter
alias_copy	✘ no converter	✘ no converter	✘ no converter	✘ no converter	✘ no converter
all	✔	✔	✔	✔	✔
allclose	✔	✔	✔	—	—
amax	✔	✔	✔	✔	✔
amin	✔	✔	✔	✔	✔
aminmax	✔	✔	✔	✔	✔
angle	✔	✔	✔	⚠ xfail	⚠ xfail
any	✔	✔	✔	✔	✔
arange	✔	✔	✔	✔	✔
argmax	✔	✔	✔	✔	✔
argmin	✔	✔	✔	✔	✔
argsort	✔	⚠ xfail	⚠ xfail	✔	✔
argwhere	✔	✔	✔	✔	✔
as_strided	✔	✔	✔	✔	✔
as_strided_copy	✔	✔	✔	✔	✔
as_strided_scatter	✔	✔	✔	✔	✔
asin	✔	✔	✔	⚠ xfail	⚠ xfail
asinh	✔	✔	✔	⚠ xfail	⚠ xfail
atan	✔	✔	✔	⚠ xfail	⚠ xfail
atan2	✔	✔	✔	⚠ xfail	⚠ xfail
atanh	✔	✔	✔	⚠ xfail	⚠ xfail
atleast_1d	✔	✔	✔	✔	✔
atleast_2d	✔	✔	✔	✔	✔
atleast_3d	✔	✔	✔	✔	✔
baddbmm	✔	✔	✔	✔	✔
bernoulli	✘ no converter	✘ no converter	✘ no converter	—	—
bfloat16	⚠ xfail	⚠ xfail	⚠ xfail	⚠ xfail	⚠ xfail
bincount	—	—	—	✔	✔
bitwise_and	—	—	—	✔	✔
bitwise_left_shift	—	—	—	⚠ xfail	⚠ xfail
bitwise_not	—	—	—	✔	✔
bitwise_or	—	—	—	✔	✔
bitwise_right_shift	—	—	—	⚠ xfail	⚠ xfail
bitwise_xor	—	—	—	⚠ xfail	⚠ xfail
block_diag	✔	✔	✔	✔	✔
bmm	✔	✔	⚠ xfail	✔	✔
bool	✔	✔	✔	✔	✔
broadcast_shapes	✔	—	—	—	—
broadcast_tensors	⚠ xfail	⚠ xfail	⚠ xfail	⚠ xfail	⚠ xfail
broadcast_to	✔	✔	✔	✔	✔
bucketize	✔	✔	✔	✔	✔
byte	✔	✔	✔	✔	✔
cartesian_prod	✔	✔	✔	✔	✔
cat	✔	✔	✔	✔	✔
cdist	✔	—	—	—	—
cdouble	✔	✔	✔	✔	✔
ceil	✔	✔	✔	⚠ xfail	⚠ xfail
cfloat	✔	✔	✔	✔	✔
chalf	✔	✔	✔	✔	✔
char	✔	✔	✔	✔	✔
cholesky	✔	—	—	—	—
cholesky_inverse	⚠ xfail	—	—	—	—
cholesky_solve	⚠ xfail	—	—	—	—
chunk	✔	✔	✔	✔	✔
clamp	⚠ xfail	⚠ xfail	⚠ xfail	⚠ xfail	⚠ xfail
clamp_max	✔	✔	✔	✔	✔
clamp_min	✔	✔	✔	✔	✔
clone	✔	✔	✔	✔	✔
column_stack	✔	✔	✔	✔	✔
combinations	✔	✔	✔	✔	✔
complex	✔	✔	—	—	—
conj	✔	✔	✔	✔	✔
conj_physical	✘ no converter	✘ no converter	✘ no converter	✘ no converter	✘ no converter
constant_pad_nd	✔	✔	✔	✔	✔
contiguous	✔	✔	✔	✔	✔
copysign	✘ no converter	✘ no converter	✘ no converter	✘ no converter	✘ no converter
corrcoef	⚠ xfail	⚠ xfail	⚠ xfail	⚠ xfail	⚠ xfail
cos	✔	✔	✔	⚠ xfail	⚠ xfail
cosh	✔	✔	✔	⚠ xfail	⚠ xfail
count_nonzero	✔	✔	✔	✔	✔
cov	⚠ xfail	⚠ xfail	⚠ xfail	⚠ xfail	⚠ xfail
cross	✘ no converter	✘ no converter	✘ no converter	✘ no converter	✘ no converter
cummax	✔	✔	✔	✔	✔
cummin	✔	✔	✔	✔	✔
cumprod	✔	✔	✔	✔	✔
cumsum	✔	✔	✔	✔	✔
cumulative_trapezoid	✘ no converter	✘ no converter	✘ no converter	✘ no converter	✘ no converter
deg2rad	✘ no converter	✘ no converter	✘ no converter	✘ no converter	✘ no converter
diag	✔	✔	✔	✔	✔
diag_embed	✔	✔	✔	✔	✔
diagflat	✘ no converter	✘ no converter	✘ no converter	✘ no converter	✘ no converter
diagonal	✘ no converter	✘ no converter	✘ no converter	✘ no converter	✘ no converter
diagonal_copy	✘ no converter	✘ no converter	✘ no converter	✘ no converter	✘ no converter
diagonal_scatter	✘ no converter	✘ no converter	✘ no converter	✘ no converter	✘ no converter
diff	✔	✔	✔	✔	✔
digamma	✘ no converter	✘ no converter	✘ no converter	✘ no converter	✘ no converter
dist	✔	✔	✔	—	—
dot	✘ no converter	✘ no converter	✘ no converter	✘ no converter	✘ no converter
double	✔	✔	✔	✔	✔
dsplit	✔	✔	✔	✔	✔
dstack	✔	✔	✔	✔	✔
einsum	✔	✔	✔	✔	✔
empty_permuted	✔	✔	✔	✔	✔
empty_strided	✔	✔	✔	✔	✔
eq	✔	✔	✔	✔	✔
equal	✔	✔	✔	✔	✔
erf	✔	✔	✔	⚠ xfail	⚠ xfail
erfc	✔	✔	✔	⚠ xfail	⚠ xfail
erfinv	✔	✔	⚠ xfail	⚠ xfail	⚠ xfail
exp	✔	✔	✔	⚠ xfail	⚠ xfail
exp2	✘ no converter	✘ no converter	✘ no converter	✘ no converter	✘ no converter
expand	✔	✔	✔	✔	✔
expand_as	✔	✔	✔	✔	✔
expand_copy	✔	✔	✔	✔	✔
expm1	✔	⚠ xfail	⚠ xfail	⚠ xfail	⚠ xfail
eye	✔	✔	✔	✔	✔
fft.fft	✔	—	—	✔	✔
fft.fft2	✔	—	—	✔	✔
fft.fftn	✔	—	—	✔	✔
fft.fftshift	✔	✔	✔	✔	✔
fft.hfft	✔	—	—	✔	✔
fft.hfft2	✔	—	—	✔	✔
fft.hfftn	✔	—	—	✔	✔
fft.ifft	✔	—	—	✔	✔
fft.ifft2	✔	—	—	✔	✔
fft.ifftn	✔	—	—	✔	✔
fft.ifftshift	✔	✔	✔	✔	✔
fft.ihfft	✔	—	—	✔	✔
fft.ihfft2	✔	—	—	✔	✔
fft.ihfftn	✔	—	—	✔	✔
fft.irfft	✔	—	—	✔	✔
fft.irfft2	✔	—	—	✔	✔
fft.irfftn	✔	—	—	✔	✔
fft.rfft	✔	—	—	✔	✔
fft.rfft2	✔	—	—	✔	✔
fft.rfftn	✔	—	—	✔	✔
fill	✔	✔	✔	✔	✔
flatten	✔	✔	✔	✔	✔
flip	✔	✔	✔	✔	✔
fliplr	✔	✔	✔	✔	✔
flipud	✔	✔	✔	✔	✔
float	✔	✔	✔	✔	✔
float_power	✔	✔	✔	✔	✔
floor	✔	✔	✔	⚠ xfail	⚠ xfail
floor_divide	✔	✔	✔	⚠ xfail	⚠ xfail
fmax	✔	✔	✔	✔	✔
fmin	✔	✔	✔	✔	✔
fmod	✔	✔	✔	✔	✔
frac	✔	✔	✔	—	—
frexp	✔	✔	⚠ xfail	—	—
full	✔	✔	✔	✔	✔
full_like	✔	✔	✔	✔	✔
gather	✔	✔	✔	✔	✔
gcd	—	—	—	⚠ xfail	⚠ xfail
ge	✔	✔	✔	✔	✔
geqrf	✔	—	—	—	—
gradient	✔	✔	✔	✔	✔
grid_sampler_2d	✔	✔	✔	—	—
grid_sampler_3d	✔	✔	✔	—	—
gt	✔	✔	✔	✔	✔
half	✔	✔	✔	✔	✔
hash_tensor	✘ no converter	✘ no converter	✘ no converter	✘ no converter	✘ no converter
heaviside	✔	✔	✔	✔	✔
histc	✔	✔	✔	—	—
histogram	✔	—	—	—	—
histogramdd	✔	—	—	—	—
hsplit	✔	✔	✔	✔	✔
hstack	✔	✔	✔	✔	✔
hypot	✘ no converter	✘ no converter	✘ no converter	—	—
i0	✘ no converter	✘ no converter	✘ no converter	✘ no converter	✘ no converter
igamma	✘ no converter	✘ no converter	✘ no converter	—	—
igammac	✘ no converter	✘ no converter	✘ no converter	—	—
imag	—	—	—	—	—
index_add	✔	✔	✔	✔	✔
index_copy	✔	✔	✔	✔	✔
index_fill	✔	✔	✔	✔	✔
index_put	✔	✔	✔	✔	✔
index_select	✔	✔	✔	✔	✔
inner	✘ no converter	✘ no converter	✘ no converter	✘ no converter	✘ no converter
int	✔	✔	✔	✔	✔
isclose	✔	✔	✔	✔	✔
isfinite	✔	✔	✔	✔	✔
isin	✔	✔	✔	✔	✔
isinf	✔	✔	✔	⚠ xfail	⚠ xfail
isnan	✔	✔	✔	⚠ xfail	⚠ xfail
isneginf	✔	✔	✔	✔	✔
isposinf	✔	✔	✔	✔	✔
isreal	✘ no converter	✘ no converter	✘ no converter	✘ no converter	✘ no converter
istft	—	—	—	—	—
item	✔	✔	✔	✔	✔
jiterator_2inputs_2outputs	⚠ xfail	⚠ xfail	⚠ xfail	⚠ xfail	⚠ xfail
jiterator_4inputs_with_extra_args	⚠ xfail	⚠ xfail	⚠ xfail	⚠ xfail	⚠ xfail
jiterator_binary	⚠ xfail	⚠ xfail	⚠ xfail	⚠ xfail	⚠ xfail
jiterator_binary_return_by_ref	⚠ xfail	⚠ xfail	⚠ xfail	⚠ xfail	⚠ xfail
jiterator_unary	⚠ xfail	⚠ xfail	⚠ xfail	⚠ xfail	⚠ xfail
kron	✘ no converter	✘ no converter	✘ no converter	✘ no converter	✘ no converter
kthvalue	✔	✔	✔	✔	✔
lcm	—	—	—	⚠ xfail	⚠ xfail
ldexp	✘ no converter	✘ no converter	✘ no converter	✘ no converter	✘ no converter
le	✔	✔	✔	✔	✔
lerp	✔	✔	✔	—	—
lgamma	✘ no converter	✘ no converter	✘ no converter	✘ no converter	✘ no converter
linalg.cholesky	✔	—	—	—	—
linalg.cholesky_ex	✔	—	—	—	—
linalg.cond	✘ no converter	—	—	—	—
linalg.cross	✔	✔	✔	✔	✔
linalg.det	✔	—	—	—	—
linalg.diagonal	✘ no converter	✘ no converter	✘ no converter	✘ no converter	✘ no converter
linalg.eig	✔	—	—	—	—
linalg.eigh	✔	—	—	—	—
linalg.eigvals	✔	—	—	—	—
linalg.eigvalsh	✔	—	—	—	—
linalg.householder_product	✘ no converter	—	—	—	—
linalg.inv	✘ no converter	—	—	—	—
linalg.inv_ex	✘ no converter	—	—	—	—
linalg.ldl_factor	✔	—	—	—	—
linalg.ldl_factor_ex	✔	—	—	—	—
linalg.ldl_solve	✔	—	—	—	—
linalg.lstsq	✔	—	—	—	—
linalg.lu	✔	—	—	—	—
linalg.lu_factor	✔	—	—	—	—
linalg.lu_factor_ex	✔	—	—	—	—
linalg.lu_solve	✔	—	—	—	—
linalg.matrix_norm	✔	✔	✔	—	—
linalg.matrix_power	✔	—	—	—	—
linalg.matrix_rank	✔	—	—	—	—
linalg.multi_dot	✔	✔	✔	✔	✔
linalg.norm	✔	✔	✔	—	—
linalg.pinv	✔	—	—	—	—
linalg.qr	✘ no converter	—	—	—	—
linalg.slogdet	✔	—	—	—	—
linalg.solve	✘ no converter	—	—	—	—
linalg.solve_ex	✘ no converter	—	—	—	—
linalg.solve_triangular	✔	—	—	—	—
linalg.svd	✔	—	—	—	—
linalg.svdvals	✘ no converter	—	—	—	—
linalg.tensorinv	✔	—	—	—	—
linalg.tensorsolve	✔	—	—	—	—
linalg.vander	✔	—	—	✔	✔
linalg.vecdot	✔	✔	✔	—	—
linalg.vector_norm	✔	✔	✔	—	—
linspace	✔	✔	✔	✔	✔
log	✔	✔	✔	⚠ xfail	⚠ xfail
log10	✔	✔	⚠ xfail	⚠ xfail	⚠ xfail
log1p	✔	✔	⚠ xfail	⚠ xfail	⚠ xfail
log2	✔	✔	⚠ xfail	⚠ xfail	⚠ xfail
log_softmax	✔	✔	✔	—	—
logaddexp	✔	✔	⚠ xfail	—	—
logaddexp2	✔	✔	⚠ xfail	—	—
logcumsumexp	✔	✔	✔	—	—
logdet	✔	—	—	—	—
logical_and	✔	✔	✔	✔	✔
logical_not	⚠ xfail	⚠ xfail	⚠ xfail	⚠ xfail	⚠ xfail
logical_or	✔	✔	✔	✔	✔
logical_xor	✔	✔	✔	✔	✔
logit	✔	✔	✔	⚠ xfail	⚠ xfail
logspace	✔	✔	✔	✔	✔
logsumexp	✔	✔	✔	✔	✔
long	✔	✔	✔	✔	✔
lt	✔	✔	✔	✔	✔
lu	✔	—	—	—	—
lu_solve	✘ no converter	—	—	—	—
lu_unpack	✘ no converter	—	—	—	—
mH	✔	✔	✔	✔	✔
mT	✔	✔	✔	✔	✔
masked.amax	✔	✔	✔	✔	✔
masked.amin	✔	✔	✔	✔	✔
masked.argmax	✔	✔	✔	✔	✔
masked.argmin	✔	✔	✔	✔	✔
masked.cumprod	✔	✔	✔	✔	✔
masked.cumsum	✔	✔	✔	✔	✔
masked.log_softmax	✔	✔	✔	—	—
masked.logaddexp	✔	✔	✔	—	—
masked.logsumexp	✔	✔	✔	✔	✔
masked.mean	✔	✔	✔	—	—
masked.median	✔	✔	✔	—	—
masked.norm	✔	✔	✔	—	—
masked.normalize	✔	✔	✔	—	—
masked.prod	✔	✔	✔	✔	✔
masked.softmax	✔	✔	✔	—	—
masked.softmin	✔	✔	✔	—	—
masked.std	✔	✔	✔	✔	✔
masked.sum	✔	✔	✔	✔	✔
masked.var	✔	✔	✔	✔	✔
masked_fill	✔	✔	✔	✔	✔
masked_scatter	✔	✔	✔	✔	✔
masked_select	✘ no converter	✘ no converter	✘ no converter	✘ no converter	✘ no converter
matmul	✔	✔	✔	✔	✔
matrix_exp	✘ no converter	✘ no converter	✘ no converter	—	—
max_pool2d_with_indices_backward	✔	✔	✔	—	—
maximum	✔	✔	✔	✔	✔
mean	✔	✔	✔	—	—
median	✘ no converter	✘ no converter	✘ no converter	✘ no converter	✘ no converter
minimum	✔	✔	✔	✔	✔
mm	✔	✔	✔	✔	✔
mode	✘ no converter	✘ no converter	✘ no converter	✘ no converter	✘ no converter
movedim	✔	✔	✔	✔	✔
msort	✘ no converter	✘ no converter	✘ no converter	✘ no converter	✘ no converter
mul	✔	✔	✔	✔	✔
multinomial	✔	✔	✔	—	—
mv	✘ no converter	✘ no converter	✘ no converter	✘ no converter	✘ no converter
nan_to_num	⚠ xfail	⚠ xfail	⚠ xfail	⚠ xfail	⚠ xfail
nanmean	✔	✔	✔	—	—
nanmedian	✘ no converter	✘ no converter	✘ no converter	✘ no converter	✘ no converter
nanquantile	✔	—	—	—	—
nansum	✔	✔	✔	⚠ xfail	⚠ xfail
narrow	✔	✔	✔	✔	✔
narrow_copy	✔	✔	✔	✔	✔
native_batch_norm	✔	✔	✔	—	—
native_dropout_backward	✔	✔	✔	✔	✔
native_layer_norm	✔	✔	✔	—	—
ne	✔	✔	✔	✔	✔
neg	✔	✔	✔	✔	✔
new_empty	✔	✔	✔	✔	✔
new_empty_strided	✔	✔	✔	✔	✔
new_full	✔	✔	✔	✔	✔
new_ones	✔	✔	✔	✔	✔
new_zeros	✔	✔	✔	✔	✔
nextafter	✘ no converter	✘ no converter	✘ no converter	—	—
nn.functional.adaptive_avg_pool1d	✔	✔	✔	—	—
nn.functional.adaptive_avg_pool2d	✔	✔	✔	—	—
nn.functional.adaptive_avg_pool3d	✔	✔	✔	—	—
nn.functional.adaptive_max_pool1d	✔	✔	✔	—	—
nn.functional.adaptive_max_pool2d	✔	✔	✔	—	—
nn.functional.adaptive_max_pool3d	✔	✔	✔	—	—
nn.functional.alpha_dropout	✔	✔	✔	—	—
nn.functional.avg_pool1d	✔	✔	✔	—	✔
nn.functional.avg_pool2d	✔	✔	✔	—	✔
nn.functional.avg_pool3d	✔	—	—	—	✔
nn.functional.batch_norm	✔	✔	✔	—	—
nn.functional.bilinear	✔	✔	✔	✔	✔
nn.functional.binary_cross_entropy	✔	✔	✔	—	—
nn.functional.binary_cross_entropy_with_logits	✔	✔	✔	—	—
nn.functional.celu	✔	✔	✔	—	—
nn.functional.channel_shuffle	✔	✔	✔	✔	✔
nn.functional.conv1d	✔	✔	✔	—	✔
nn.functional.conv2d	✔	✔	✔	—	✔
nn.functional.conv3d	✔	✔	✔	—	✔
nn.functional.conv_transpose1d	✔	✔	✔	—	✔
nn.functional.conv_transpose2d	✔	✔	✔	—	✔
nn.functional.conv_transpose3d	✔	✔	✔	—	✔
nn.functional.cosine_embedding_loss	✔	✔	✔	✔	✔
nn.functional.cosine_similarity	✔	✔	✔	—	—
nn.functional.cross_entropy	✔	✔	✔	—	—
nn.functional.ctc_loss	✔	—	—	—	—
nn.functional.dropout	✔	✔	✔	—	—
nn.functional.dropout2d	✔	✔	✔	—	—
nn.functional.dropout3d	✔	✔	✔	—	—
nn.functional.elu	✔	✔	✔	—	—
nn.functional.embedding	✔	✔	✔	—	—
nn.functional.embedding_bag	✔	✔	✔	—	—
nn.functional.fractional_max_pool2d	✔	✔	✔	—	—
nn.functional.fractional_max_pool3d	✔	✔	✔	—	—
nn.functional.gaussian_nll_loss	✔	✔	✔	—	—
nn.functional.gelu	✔	✔	✔	—	—
nn.functional.glu	✔	✔	✔	—	—
nn.functional.grid_sample	✔	✔	✔	—	—
nn.functional.group_norm	✔	✔	✔	—	—
nn.functional.hardshrink	✔	✔	✔	—	—
nn.functional.hardsigmoid	✔	✔	✔	—	—
nn.functional.hardswish	✔	✔	✔	—	—
nn.functional.hardtanh	✔	✔	✔	✔	✔
nn.functional.hinge_embedding_loss	✔	✔	✔	—	—
nn.functional.huber_loss	✔	✔	✔	—	—
nn.functional.instance_norm	✔	✔	✔	—	—
nn.functional.kl_div	✔	✔	✔	—	—
nn.functional.l1_loss	✔	✔	✔	—	—
nn.functional.layer_norm	✔	✔	✔	—	—
nn.functional.leaky_relu	✔	✔	✔	—	—
nn.functional.linear	✔	✔	✔	⚠ xfail	⚠ xfail
nn.functional.local_response_norm	✔	✔	✔	—	✔
nn.functional.logsigmoid	✔	✔	✔	—	—
nn.functional.margin_ranking_loss	✔	✔	✔	✔	✔
nn.functional.max_pool1d	✔	✔	✔	—	—
nn.functional.max_pool2d	✔	✔	✔	✔	✔
nn.functional.max_pool3d	✔	✔	✔	✔	✔
nn.functional.max_unpool1d	✔	✔	✔	—	—
nn.functional.max_unpool2d	✔	✔	✔	—	—
nn.functional.max_unpool3d	✔	✔	✔	—	—
nn.functional.mish	✔	✔	✔	—	—
nn.functional.mse_loss	✔	✔	✔	—	—
nn.functional.multi_head_attention_forward	✔	✔	✔	—	—
nn.functional.multi_margin_loss	✘ no converter	—	—	—	—
nn.functional.multilabel_margin_loss	✘ no converter	—	—	—	—
nn.functional.multilabel_soft_margin_loss	✘ no converter	✘ no converter	✘ no converter	—	—
nn.functional.nll_loss	✔	✔	✔	—	—
nn.functional.normalize	✔	✔	✔	—	—
nn.functional.one_hot	—	—	—	—	✔
nn.functional.pairwise_distance	✔	✔	✔	✔	✔
nn.functional.pdist	✘ no converter	—	—	—	—
nn.functional.pixel_shuffle	✔	✔	✔	✔	✔
nn.functional.pixel_unshuffle	✔	✔	✔	✔	✔
nn.functional.poisson_nll_loss	✔	✔	✔	✔	✔
nn.functional.prelu	✔	✔	✔	—	—
nn.functional.relu	✔	✔	✔	⚠ xfail	⚠ xfail
nn.functional.relu6	✔	✔	✔	✔	✔
nn.functional.rms_norm	✔	✔	✔	—	—
nn.functional.rrelu	✔	—	✔	—	—
nn.functional.scaled_dot_product_attention	✔	✔	✔	—	—
nn.functional.selu	✔	✔	✔	—	—
nn.functional.silu	✔	✔	✔	—	—
nn.functional.smooth_l1_loss	✔	✔	✔	—	—
nn.functional.soft_margin_loss	✔	✔	✔	—	—
nn.functional.softmin	✔	✔	✔	—	—
nn.functional.softplus	✔	✔	✔	—	—
nn.functional.softshrink	✔	✔	✔	—	—
nn.functional.softsign	✔	✔	✔	⚠ xfail	⚠ xfail
nn.functional.tanhshrink	✔	✔	✔	⚠ xfail	⚠ xfail
nn.functional.threshold	✔	✔	✔	✔	✔
nn.functional.triplet_margin_loss	✔	✔	✔	✔	✔
nn.functional.triplet_margin_with_distance_loss	✔	✔	✔	✔	✔
nn.functional.unfold	✔	✔	✔	—	—
nn.functional.upsample_bilinear	✔	✔	✔	—	—
nn.functional.upsample_nearest	✔	✔	✔	—	—
nonzero	✔	✔	✔	✔	✔
nonzero_static	✔	✔	✔	✔	✔
norm	✔	✔	✔	—	—
ones	✔	✔	✔	✔	✔
ones_like	✔	✔	✔	✔	✔
ormqr	✔	—	—	—	—
outer	✔	✔	✔	✔	✔
pca_lowrank	✔	—	—	—	—
permute	✔	✔	✔	✔	✔
permute_copy	✔	✔	✔	✔	✔
pinverse	✘ no converter	—	—	—	—
polar	✔	—	—	—	—
positive	✘ no converter	✘ no converter	✘ no converter	✘ no converter	✘ no converter
pow	✔	✔	✔	✔	✔
prod	✔	✔	✔	⚠ xfail	✔
put	✔	✔	✔	✔	✔
qr	✘ no converter	—	—	—	—
quantile	✔	—	—	—	—
rad2deg	✘ no converter	✘ no converter	✘ no converter	✘ no converter	✘ no converter
randint_like	✔	✔	✔	✔	✔
ravel	✔	✔	✔	✔	✔
real	✘ no converter	✘ no converter	✘ no converter	✘ no converter	✘ no converter
reciprocal	✔	✔	✔	⚠ xfail	⚠ xfail
remainder	✔	✔	✔	✔	✔
renorm	✔	✔	✔	—	—
repeat	✔	✔	✔	✔	✔
repeat_interleave	✔	✔	✔	✔	✔
reshape	✔	✔	✔	✔	✔
reshape_as	✔	✔	✔	✔	✔
resize_	✔	✔	✔	✔	✔
resize_as_	✘ no converter	✘ no converter	✘ no converter	✘ no converter	✘ no converter
resolve_conj	✘ no converter	✘ no converter	✘ no converter	✘ no converter	✘ no converter
resolve_neg	✘ no converter	✘ no converter	✘ no converter	✘ no converter	✘ no converter
roll	✔	✔	✔	✔	✔
rot90	✔	✔	✔	✔	✔
round	✔	✔	✔	⚠ xfail	⚠ xfail
rsqrt	✔	✔	✔	⚠ xfail	⚠ xfail
rsub	✘ no converter	✘ no converter	✘ no converter	✘ no converter	✘ no converter
scalar_tensor	✔	✔	✔	✔	✔
scatter	✔	✔	✔	✔	✔
scatter_add	✔	✔	✔	✔	✔
searchsorted	✔	✔	✔	✔	✔
select	✔	✔	✔	✔	✔
select_scatter	✔	✔	✔	✔	✔
sgn	✘ no converter	✘ no converter	✘ no converter	✘ no converter	✘ no converter
short	✔	✔	✔	✔	✔
sigmoid	✔	✔	✔	⚠ xfail	⚠ xfail
sign	✔	✔	✔	✔	✔
signal.windows.bartlett	✔	—	—	—	—
signal.windows.blackman	✔	—	—	—	—
signal.windows.cosine	✔	—	—	—	—
signal.windows.exponential	✔	—	—	—	—
signal.windows.gaussian	✔	—	—	—	—
signal.windows.general_cosine	✔	—	—	—	—
signal.windows.general_hamming	✔	—	—	—	—
signal.windows.hamming	✔	—	—	—	—
signal.windows.hann	✔	—	—	—	—
signal.windows.kaiser	✔	—	—	—	—
signal.windows.nuttall	✔	—	—	—	—
signbit	✔	✔	✔	✔	✔
sin	✔	✔	✔	⚠ xfail	⚠ xfail
sinc	✘ no converter	✘ no converter	✘ no converter	✘ no converter	✘ no converter
sinh	✔	✔	✔	⚠ xfail	⚠ xfail
slice	✔	✔	✔	✔	✔
slice_scatter	✔	✔	✔	✔	✔
softmax	✔	✔	✔	—	—
sort	✘ no converter	✘ no converter	✘ no converter	✘ no converter	✘ no converter
sparse.sampled_addmm	✔	—	—	—	—
special.airy_ai	✘ no converter	—	—	✘ no converter	✘ no converter
special.bessel_j0	✘ no converter	—	—	✘ no converter	✘ no converter
special.bessel_j1	✘ no converter	—	—	✘ no converter	✘ no converter
special.bessel_y0	✘ no converter	—	—	✘ no converter	✘ no converter
special.bessel_y1	✘ no converter	—	—	✘ no converter	✘ no converter
special.chebyshev_polynomial_t	✘ no converter	—	—	✘ no converter	✘ no converter
special.chebyshev_polynomial_u	✘ no converter	—	—	✘ no converter	✘ no converter
special.chebyshev_polynomial_v	✘ no converter	—	—	✘ no converter	✘ no converter
special.chebyshev_polynomial_w	✘ no converter	—	—	✘ no converter	✘ no converter
special.entr	✘ no converter	✘ no converter	✘ no converter	✘ no converter	✘ no converter
special.erfcx	✘ no converter	—	—	✘ no converter	✘ no converter
special.hermite_polynomial_h	✘ no converter	—	—	✘ no converter	✘ no converter
special.hermite_polynomial_he	✘ no converter	—	—	✘ no converter	✘ no converter
special.i0e	✘ no converter	✘ no converter	✘ no converter	✘ no converter	✘ no converter
special.i1	✘ no converter	✘ no converter	✘ no converter	✘ no converter	✘ no converter
special.i1e	✘ no converter	✘ no converter	✘ no converter	✘ no converter	✘ no converter
special.laguerre_polynomial_l	✘ no converter	—	—	✘ no converter	✘ no converter
special.legendre_polynomial_p	✘ no converter	—	—	✘ no converter	✘ no converter
special.log_ndtr	✘ no converter	—	—	✘ no converter	✘ no converter
special.modified_bessel_i0	✘ no converter	—	—	✘ no converter	✘ no converter
special.modified_bessel_i1	✘ no converter	—	—	✘ no converter	✘ no converter
special.modified_bessel_k0	✘ no converter	—	—	✘ no converter	✘ no converter
special.modified_bessel_k1	✘ no converter	—	—	✘ no converter	✘ no converter
special.ndtr	✘ no converter	✘ no converter	✘ no converter	✘ no converter	✘ no converter
special.ndtri	✘ no converter	—	—	✘ no converter	✘ no converter
special.scaled_modified_bessel_k0	✘ no converter	—	—	✘ no converter	✘ no converter
special.scaled_modified_bessel_k1	✘ no converter	—	—	✘ no converter	✘ no converter
special.shifted_chebyshev_polynomial_t	✘ no converter	—	—	✘ no converter	✘ no converter
special.shifted_chebyshev_polynomial_u	✘ no converter	—	—	✘ no converter	✘ no converter
special.shifted_chebyshev_polynomial_v	✘ no converter	—	—	✘ no converter	✘ no converter
special.shifted_chebyshev_polynomial_w	✘ no converter	—	—	✘ no converter	✘ no converter
special.spherical_bessel_j0	✘ no converter	—	—	✘ no converter	✘ no converter
special.xlog1py	✘ no converter	✘ no converter	✘ no converter	✘ no converter	✘ no converter
special.zeta	✘ no converter	—	—	✘ no converter	✘ no converter
split	✔	✔	✔	✔	✔
split_with_sizes	✔	✔	✔	✔	✔
split_with_sizes_copy	✔	✔	✔	✔	✔
sqrt	✔	✔	✔	⚠ xfail	⚠ xfail
square	✔	✔	✔	✔	✔
squeeze	✔	✔	✔	✔	✔
squeeze_copy	✘ no converter	✘ no converter	✘ no converter	✘ no converter	✘ no converter
stack	✔	✔	✔	✔	✔
std	✔	✔	✔	—	—
std_mean	✔	✔	✔	—	—
stft	✔	—	—	—	—
sub	✔	✔	✔	✔	✔
sum	✔	✔	✔	⚠ xfail	✔
sum_to_size	✔	✔	✔	✔	✔
svd	✔	—	—	—	—
svd_lowrank	✔	—	—	—	—
t	✔	✔	✔	✔	✔
t_copy	✘ no converter	✘ no converter	✘ no converter	✘ no converter	✘ no converter
take	✔	✔	✔	✔	✔
take_along_dim	✔	✔	✔	✔	✔
tan	✔	✔	✔	⚠ xfail	⚠ xfail
tanh	✔	✔	✔	⚠ xfail	⚠ xfail
tensor_split	✔	✔	✔	✔	✔
tensordot	✔	✔	✔	✔	✔
tile	✔	✔	✔	✔	✔
to	✔	✔	✔	✔	✔
to_sparse	✘ no converter	✘ no converter	✘ no converter	✘ no converter	✘ no converter
topk	✔	✔	✔	✔	✔
torch._scaled_mm	—	—	—	—	—
torch._scaled_mm_v2	—	—	—	—	—
torch.ops.aten._efficient_attention_forward	—	—	—	—	—
torch.ops.aten._flash_attention_forward	—	—	—	—	—
torch.ops.aten._safe_softmax.default	✔	✔	✔	✔	✔
trace	✘ no converter	—	—	✘ no converter	✘ no converter
transpose	✔	✔	✔	✔	✔
transpose_copy	✔	✔	✔	✔	✔
trapezoid	✘ no converter	✘ no converter	✘ no converter	✘ no converter	✘ no converter
trapz	✘ no converter	✘ no converter	✘ no converter	✘ no converter	✘ no converter
triangular_solve	✘ no converter	—	—	—	—
tril	✔	✔	✔	⚠ xfail	✔
tril_indices	—	—	—	✔	✔
triu	✔	✔	✔	⚠ xfail	✔
triu_indices	—	—	—	✔	✔
true_divide	✔	✔	✔	✔	✔
trunc	✔	✔	✔	⚠ xfail	⚠ xfail
unbind	✔	✔	✔	✔	✔
unbind_copy	✔	✔	✔	✔	✔
unflatten	✔	✔	✔	✔	✔
unfold	✔	✔	✔	✔	✔
unfold_copy	✔	✔	✔	✔	✔
unique	✔	✔	✔	✔	✔
unique_consecutive	✔	✔	✔	✔	✔
unravel_index	—	—	—	✔	✔
unsafe_chunk	✔	✔	✔	✔	✔
unsafe_split	✔	✔	✔	✔	✔
unsqueeze	✔	✔	✔	✔	✔
unsqueeze_copy	✔	✔	✔	✔	✔
var	✘ no converter	✘ no converter	✘ no converter	—	—
var_mean	✘ no converter	✘ no converter	✘ no converter	—	—
vdot	✘ no converter	✘ no converter	✘ no converter	✘ no converter	✘ no converter
view	✔	✔	✔	✔	✔
view_as	✘ no converter	✘ no converter	✘ no converter	✘ no converter	✘ no converter
view_as_complex	✔	✔	—	—	—
view_as_real	—	—	—	—	—
view_copy	✔	✔	✔	✔	✔
vsplit	✔	✔	✔	✔	✔
vstack	✔	✔	✔	✔	✔
where	✔	✔	✔	✔	✔
xlogy	✔	✔	✔	⚠ xfail	⚠ xfail
zero_	✘ no converter	✘ no converter	✘ no converter	✘ no converter	✘ no converter
zeros	✔	✔	✔	✔	✔
zeros_like	✔	✔	✔	✔	✔

[runpythonerror] W0429 17:12:59.975000 18809 torch/_native/cutedsl_utils.py:55] CuTeDSL operators require optional Python packages nvidia-cutlass-dsl and apache-tvm-ffi; missing optional dependency nvidia_cutlass_dsl (importlib.util.find_spec(nvidia_cutlass_dsl) failed)

Torch tracing export path

Torch tracing export path (tracing=True)

Op	float32	float16	bfloat16	int32	int64
H	✘ no converter	✘ no converter	✘ no converter	✘ no converter	✘ no converter
T	✔	✔	✔	✔	✔
__getitem__	✔	✔	✔	✔	✔
__radd__	⚠ xfail	⚠ xfail	⚠ xfail	⚠ xfail	⚠ xfail
__rand__	—	—	—	⚠ xfail	⚠ xfail
__rdiv__	⚠ xfail	⚠ xfail	⚠ xfail	⚠ xfail	⚠ xfail
__rmatmul__	✔	✔	✔	✔	✔
__rmod__	✔	✔	✔	—	—
__rmul__	⚠ xfail	⚠ xfail	⚠ xfail	⚠ xfail	⚠ xfail
__ror__	—	—	—	⚠ xfail	⚠ xfail
__rpow__	✔	✔	✔	⚠ xfail	⚠ xfail
__rsub__	✘ no converter	✘ no converter	✘ no converter	✘ no converter	✘ no converter
__rxor__	—	—	—	⚠ xfail	⚠ xfail
_batch_norm_with_update	✔	✔	✔	—	—
_chunk_cat	✔	✔	✔	✔	✔
_native_batch_norm_legit	✔	✔	✔	—	—
_softmax_backward_data	✔	✔	✔	—	—
_unsafe_masked_index	✔	✔	✔	✔	✔
_unsafe_masked_index_put_accumulate	✔	✔	✔	✔	✔
_upsample_bilinear2d_aa	✔	—	—	—	—
abs	✔	✔	✔	✔	✔
acos	✔	✔	✔	⚠ xfail	⚠ xfail
acosh	✔	✔	✔	⚠ xfail	⚠ xfail
add	✔	✔	✔	✔	✔
addbmm	✔	✔	✔	✔	✔
addcdiv	✘ no converter	✘ no converter	✘ no converter	—	—
addcmul	✔	⚠ xfail	⚠ xfail	✔	✔
addmm	✔	✔	✔	✔	✔
addmv	✔	✔	✔	✔	✔
addr	✘ no converter	✘ no converter	✘ no converter	✘ no converter	✘ no converter
alias_copy	✘ no converter	✘ no converter	✘ no converter	✘ no converter	✘ no converter
all	✔	✔	✔	✔	✔
allclose	✔	✔	✔	—	—
amax	✔	✔	✔	✔	✔
amin	✔	✔	✔	✔	✔
aminmax	✔	✔	✔	✔	✔
angle	✔	✔	✔	⚠ xfail	⚠ xfail
any	✔	✔	✔	✔	✔
arange	✔	✔	✔	✔	✔
argmax	✔	✔	✔	✔	✔
argmin	✔	✔	✔	✔	✔
argsort	✔	⚠ xfail	⚠ xfail	✔	✔
argwhere	✔	✔	✔	✔	✔
as_strided	✔	✔	✔	✔	✔
as_strided_copy	✔	✔	✔	✔	✔
as_strided_scatter	✔	✔	✔	✔	✔
asin	✔	✔	✔	⚠ xfail	⚠ xfail
asinh	✔	✔	✔	⚠ xfail	⚠ xfail
atan	✔	✔	✔	⚠ xfail	⚠ xfail
atan2	✔	✔	✔	⚠ xfail	⚠ xfail
atanh	✔	✔	✔	⚠ xfail	⚠ xfail
atleast_1d	✔	✔	✔	✔	✔
atleast_2d	✔	✔	✔	✔	✔
atleast_3d	✔	✔	✔	✔	✔
baddbmm	✔	✔	✔	✔	✔
bernoulli	✘ no converter	✘ no converter	✘ no converter	—	—
bfloat16	⚠ xfail	⚠ xfail	⚠ xfail	⚠ xfail	⚠ xfail
bincount	—	—	—	✔	✔
bitwise_and	—	—	—	⚠ xfail	⚠ xfail
bitwise_left_shift	—	—	—	⚠ xfail	⚠ xfail
bitwise_not	—	—	—	✔	✔
bitwise_or	—	—	—	⚠ xfail	⚠ xfail
bitwise_right_shift	—	—	—	⚠ xfail	⚠ xfail
bitwise_xor	—	—	—	⚠ xfail	⚠ xfail
block_diag	✔	✔	✔	✔	✔
bmm	✔	✔	⚠ xfail	✔	✔
bool	✔	✔	✔	✔	✔
broadcast_shapes	✔	—	—	—	—
broadcast_tensors	⚠ xfail	⚠ xfail	⚠ xfail	⚠ xfail	⚠ xfail
broadcast_to	✔	✔	✔	✔	✔
bucketize	✔	✔	✔	✔	✔
byte	✔	✔	✔	✔	✔
cartesian_prod	✔	✔	✔	✔	✔
cat	✔	✔	✔	✔	✔
cdist	✔	—	—	—	—
cdouble	✔	✔	✔	✔	✔
ceil	✔	✔	✔	⚠ xfail	⚠ xfail
cfloat	✔	✔	✔	✔	✔
chalf	✔	✔	✔	✔	✔
char	✔	✔	✔	✔	✔
cholesky	✔	—	—	—	—
cholesky_inverse	⚠ xfail	—	—	—	—
cholesky_solve	⚠ xfail	—	—	—	—
chunk	✔	✔	✔	✔	✔
clamp	⚠ xfail	⚠ xfail	⚠ xfail	⚠ xfail	⚠ xfail
clamp_max	✔	✔	✔	✔	✔
clamp_min	✔	✔	✔	✔	✔
clone	✔	✔	✔	✔	✔
column_stack	✔	✔	✔	✔	✔
combinations	✔	✔	✔	✔	✔
complex	✔	✔	—	—	—
conj	⚠ xfail	⚠ xfail	⚠ xfail	⚠ xfail	⚠ xfail
conj_physical	✘ no converter	✘ no converter	✘ no converter	✘ no converter	✘ no converter
constant_pad_nd	✔	✔	✔	✔	✔
contiguous	✔	✔	✔	✔	✔
copysign	✘ no converter	✘ no converter	✘ no converter	✘ no converter	✘ no converter
corrcoef	⚠ xfail	⚠ xfail	⚠ xfail	⚠ xfail	⚠ xfail
cos	✔	✔	✔	⚠ xfail	⚠ xfail
cosh	✔	✔	✔	⚠ xfail	⚠ xfail
count_nonzero	✔	✔	✔	✔	✔
cov	⚠ xfail	⚠ xfail	⚠ xfail	⚠ xfail	⚠ xfail
cross	✘ no converter	✘ no converter	✘ no converter	✘ no converter	✘ no converter
cummax	✔	✔	✔	✔	✔
cummin	✔	✔	✔	✔	✔
cumprod	✔	✔	✔	✔	✔
cumsum	✔	✔	✔	✔	✔
cumulative_trapezoid	✘ no converter	✘ no converter	✘ no converter	✘ no converter	✘ no converter
deg2rad	✘ no converter	✘ no converter	✘ no converter	✘ no converter	✘ no converter
diag	✔	✔	✔	✔	✔
diag_embed	✔	✔	✔	✔	✔
diagflat	✘ no converter	✘ no converter	✘ no converter	✘ no converter	✘ no converter
diagonal	✘ no converter	✘ no converter	✘ no converter	✘ no converter	✘ no converter
diagonal_copy	✘ no converter	✘ no converter	✘ no converter	✘ no converter	✘ no converter
diagonal_scatter	✘ no converter	✘ no converter	✘ no converter	✘ no converter	✘ no converter
diff	✔	✔	✔	✔	✔
digamma	✘ no converter	✘ no converter	✘ no converter	✘ no converter	✘ no converter
dist	✔	✔	✔	—	—
dot	✘ no converter	✘ no converter	✘ no converter	✘ no converter	✘ no converter
double	✔	✔	✔	✔	✔
dsplit	✔	✔	✔	✔	✔
dstack	✔	✔	✔	✔	✔
einsum	✔	✔	✔	✔	✔
empty_permuted	✔	✔	✔	✔	✔
empty_strided	✔	✔	✔	✔	✔
eq	✔	✔	✔	✔	✔
equal	✔	✔	✔	✔	✔
erf	✔	✔	✔	⚠ xfail	⚠ xfail
erfc	✔	✔	✔	⚠ xfail	⚠ xfail
erfinv	✔	✔	⚠ xfail	⚠ xfail	⚠ xfail
exp	✔	✔	✔	⚠ xfail	⚠ xfail
exp2	✘ no converter	✘ no converter	✘ no converter	✘ no converter	✘ no converter
expand	✔	✔	✔	✔	✔
expand_as	✔	✔	✔	✔	✔
expand_copy	✔	✔	✔	✔	✔
expm1	✔	⚠ xfail	⚠ xfail	⚠ xfail	⚠ xfail
eye	✔	✔	✔	✔	✔
fft.fft	✔	—	—	✔	✔
fft.fft2	✔	—	—	✔	✔
fft.fftn	✔	—	—	✔	✔
fft.fftshift	✔	✔	✔	✔	✔
fft.hfft	✔	—	—	✔	✔
fft.hfft2	✔	—	—	✔	✔
fft.hfftn	✔	—	—	✔	✔
fft.ifft	✔	—	—	✔	✔
fft.ifft2	✔	—	—	✔	✔
fft.ifftn	✔	—	—	✔	✔
fft.ifftshift	✔	✔	✔	✔	✔
fft.ihfft	✔	—	—	✔	✔
fft.ihfft2	✔	—	—	✔	✔
fft.ihfftn	✔	—	—	✔	✔
fft.irfft	✔	—	—	✔	✔
fft.irfft2	✔	—	—	✔	✔
fft.irfftn	✔	—	—	✔	✔
fft.rfft	✔	—	—	✔	✔
fft.rfft2	✔	—	—	✔	✔
fft.rfftn	✔	—	—	✔	✔
fill	✔	✔	✔	✔	✔
flatten	⚠ xfail	⚠ xfail	⚠ xfail	⚠ xfail	⚠ xfail
flip	✔	✔	✔	✔	✔
fliplr	✔	✔	✔	✔	✔
flipud	✔	✔	✔	✔	✔
float	✔	✔	✔	✔	✔
float_power	✔	✔	⚠ xfail	✔	✔
floor	✔	✔	✔	⚠ xfail	⚠ xfail
floor_divide	✔	✔	✔	⚠ xfail	⚠ xfail
fmax	✔	✔	✔	✔	✔
fmin	✔	✔	✔	✔	✔
fmod	✔	✔	✔	✔	✔
frac	✔	✔	✔	—	—
frexp	✔	✔	⚠ xfail	—	—
full	✔	✔	✔	✔	✔
full_like	✔	✔	✔	✔	✔
gather	✔	✔	✔	✔	✔
gcd	—	—	—	⚠ xfail	⚠ xfail
ge	✔	✔	✔	✔	✔
geqrf	✔	—	—	—	—
gradient	✔	✔	✔	✔	✔
grid_sampler_2d	✔	✔	✔	—	—
grid_sampler_3d	✔	✔	✔	—	—
gt	✔	✔	✔	✔	✔
half	✔	✔	✔	✔	✔
hash_tensor	✘ no converter	✘ no converter	✘ no converter	✘ no converter	✘ no converter
heaviside	✔	✔	✔	✔	✔
histc	✔	✔	✔	—	—
histogram	✔	—	—	—	—
histogramdd	✔	—	—	—	—
hsplit	✔	✔	✔	✔	✔
hstack	✔	✔	✔	✔	✔
hypot	✘ no converter	✘ no converter	✘ no converter	—	—
i0	✘ no converter	✘ no converter	✘ no converter	✘ no converter	✘ no converter
igamma	✘ no converter	✘ no converter	✘ no converter	—	—
igammac	✘ no converter	✘ no converter	✘ no converter	—	—
imag	—	—	—	—	—
index_add	✔	✔	✔	✔	✔
index_copy	✔	✔	✔	✔	✔
index_fill	✔	✔	✔	✔	✔
index_put	✔	✔	✔	✔	✔
index_select	✔	✔	✔	✔	✔
inner	✘ no converter	✘ no converter	✘ no converter	✘ no converter	✘ no converter
int	✔	✔	✔	✔	✔
isclose	✔	✔	✔	✔	✔
isfinite	✔	✔	✔	✔	✔
isin	✔	✔	✔	✔	✔
isinf	✔	✔	✔	⚠ xfail	⚠ xfail
isnan	✔	✔	✔	⚠ xfail	⚠ xfail
isneginf	✔	✔	✔	✔	✔
isposinf	✔	✔	✔	✔	✔
isreal	✘ no converter	✘ no converter	✘ no converter	✘ no converter	✘ no converter
istft	—	—	—	—	—
item	✔	✔	✔	✔	✔
jiterator_2inputs_2outputs	⚠ xfail	⚠ xfail	⚠ xfail	⚠ xfail	⚠ xfail
jiterator_4inputs_with_extra_args	⚠ xfail	⚠ xfail	⚠ xfail	⚠ xfail	⚠ xfail
jiterator_binary	⚠ xfail	⚠ xfail	⚠ xfail	⚠ xfail	⚠ xfail
jiterator_binary_return_by_ref	⚠ xfail	⚠ xfail	⚠ xfail	⚠ xfail	⚠ xfail
jiterator_unary	⚠ xfail	⚠ xfail	⚠ xfail	⚠ xfail	⚠ xfail
kron	✘ no converter	✘ no converter	✘ no converter	✘ no converter	✘ no converter
kthvalue	✔	✔	✔	✔	✔
lcm	—	—	—	⚠ xfail	⚠ xfail
ldexp	✘ no converter	✘ no converter	✘ no converter	✘ no converter	✘ no converter
le	✔	✔	✔	✔	✔
lerp	✔	✔	✔	—	—
lgamma	✘ no converter	✘ no converter	✘ no converter	✘ no converter	✘ no converter
linalg.cholesky	✔	—	—	—	—
linalg.cholesky_ex	✔	—	—	—	—
linalg.cond	✘ no converter	—	—	—	—
linalg.cross	✔	✔	✔	✔	✔
linalg.det	✔	—	—	—	—
linalg.diagonal	✘ no converter	✘ no converter	✘ no converter	✘ no converter	✘ no converter
linalg.eig	✔	—	—	—	—
linalg.eigh	✔	—	—	—	—
linalg.eigvals	✔	—	—	—	—
linalg.eigvalsh	✔	—	—	—	—
linalg.householder_product	✘ no converter	—	—	—	—
linalg.inv	✘ no converter	—	—	—	—
linalg.inv_ex	✘ no converter	—	—	—	—
linalg.ldl_factor	✔	—	—	—	—
linalg.ldl_factor_ex	✔	—	—	—	—
linalg.ldl_solve	✔	—	—	—	—
linalg.lstsq	✔	—	—	—	—
linalg.lu	✔	—	—	—	—
linalg.lu_factor	✔	—	—	—	—
linalg.lu_factor_ex	✔	—	—	—	—
linalg.lu_solve	✔	—	—	—	—
linalg.matrix_norm	✔	✔	✔	—	—
linalg.matrix_power	✔	—	—	—	—
linalg.matrix_rank	✔	—	—	—	—
linalg.multi_dot	✔	✔	✔	✔	✔
linalg.norm	✔	✔	✔	—	—
linalg.pinv	✔	—	—	—	—
linalg.qr	✘ no converter	—	—	—	—
linalg.slogdet	✔	—	—	—	—
linalg.solve	✘ no converter	—	—	—	—
linalg.solve_ex	✘ no converter	—	—	—	—
linalg.solve_triangular	✔	—	—	—	—
linalg.svd	✔	—	—	—	—
linalg.svdvals	✘ no converter	—	—	—	—
linalg.tensorinv	✔	—	—	—	—
linalg.tensorsolve	✔	—	—	—	—
linalg.vander	✔	—	—	✔	✔
linalg.vecdot	✔	✔	✔	—	—
linalg.vector_norm	✔	✔	✔	—	—
linspace	✔	✔	✔	✔	✔
log	✔	✔	✔	⚠ xfail	⚠ xfail
log10	✔	✔	⚠ xfail	⚠ xfail	⚠ xfail
log1p	✔	✔	⚠ xfail	⚠ xfail	⚠ xfail
log2	✔	✔	⚠ xfail	⚠ xfail	⚠ xfail
log_softmax	✔	✔	✔	—	—
logaddexp	✔	✔	⚠ xfail	—	—
logaddexp2	✔	✔	⚠ xfail	—	—
logcumsumexp	✔	✔	✔	—	—
logdet	✔	—	—	—	—
logical_and	⚠ xfail	⚠ xfail	⚠ xfail	⚠ xfail	⚠ xfail
logical_not	⚠ xfail	⚠ xfail	⚠ xfail	⚠ xfail	⚠ xfail
logical_or	⚠ xfail	⚠ xfail	⚠ xfail	⚠ xfail	⚠ xfail
logical_xor	⚠ xfail	⚠ xfail	⚠ xfail	⚠ xfail	⚠ xfail
logit	✔	✔	✔	⚠ xfail	⚠ xfail
logspace	✔	✔	✔	✔	✔
logsumexp	✔	✔	✔	✔	✔
long	✔	✔	✔	✔	✔
lt	✔	✔	✔	✔	✔
lu	✔	—	—	—	—
lu_solve	✘ no converter	—	—	—	—
lu_unpack	✘ no converter	—	—	—	—
mH	✔	✔	✔	✔	✔
mT	✔	✔	✔	✔	✔
masked.amax	✔	✔	✔	✔	✔
masked.amin	✔	✔	✔	✔	✔
masked.argmax	✔	✔	✔	✔	✔
masked.argmin	✔	✔	✔	✔	✔
masked.cumprod	✔	✔	✔	✔	✔
masked.cumsum	✔	✔	✔	✔	✔
masked.log_softmax	✔	✔	✔	—	—
masked.logaddexp	✔	✔	✔	—	—
masked.logsumexp	✔	✔	✔	✔	✔
masked.mean	✔	✔	✔	—	—
masked.median	✔	✔	✔	—	—
masked.norm	✔	✔	✔	—	—
masked.normalize	✔	✔	✔	—	—
masked.prod	✔	✔	✔	✔	✔
masked.softmax	✔	✔	✔	—	—
masked.softmin	✔	✔	✔	—	—
masked.std	✔	✔	✔	✔	✔
masked.sum	✔	✔	✔	✔	✔
masked.var	✔	✔	✔	✔	✔
masked_fill	✔	✔	✔	✔	✔
masked_scatter	✔	✔	✔	✔	✔
masked_select	✘ no converter	✘ no converter	✘ no converter	✘ no converter	✘ no converter
matmul	✔	✔	✔	✔	✔
matrix_exp	✘ no converter	✘ no converter	✘ no converter	—	—
max_pool2d_with_indices_backward	✔	✔	✔	—	—
maximum	✔	✔	✔	✔	✔
mean	✔	✔	✔	—	—
median	✘ no converter	✘ no converter	✘ no converter	✘ no converter	✘ no converter
minimum	✔	✔	✔	✔	✔
mm	✔	✔	✔	✔	✔
mode	✘ no converter	✘ no converter	✘ no converter	✘ no converter	✘ no converter
movedim	✔	✔	✔	✔	✔
msort	✘ no converter	✘ no converter	✘ no converter	✘ no converter	✘ no converter
mul	✔	✔	✔	✔	✔
multinomial	✔	✔	✔	—	—
mv	✘ no converter	✘ no converter	✘ no converter	✘ no converter	✘ no converter
nan_to_num	⚠ xfail	⚠ xfail	⚠ xfail	⚠ xfail	⚠ xfail
nanmean	✔	✔	✔	—	—
nanmedian	✘ no converter	✘ no converter	✘ no converter	✘ no converter	✘ no converter
nanquantile	✔	—	—	—	—
nansum	✔	✔	✔	⚠ xfail	⚠ xfail
narrow	✔	✔	✔	✔	✔
narrow_copy	✔	✔	✔	✔	✔
native_batch_norm	✔	✔	✔	—	—
native_dropout_backward	✔	✔	✔	✔	✔
native_layer_norm	✔	✔	✔	—	—
ne	✔	✔	✔	✔	✔
neg	✔	✔	✔	✔	✔
new_empty	✔	✔	✔	✔	✔
new_empty_strided	✔	✔	✔	✔	✔
new_full	✔	✔	✔	✔	✔
new_ones	✔	✔	✔	✔	✔
new_zeros	✔	✔	✔	✔	✔
nextafter	✘ no converter	✘ no converter	✘ no converter	—	—
nn.functional.adaptive_avg_pool1d	✔	✔	✔	—	—
nn.functional.adaptive_avg_pool2d	✔	✔	✔	—	—
nn.functional.adaptive_avg_pool3d	✔	✔	✔	—	—
nn.functional.adaptive_max_pool1d	✔	✔	✔	—	—
nn.functional.adaptive_max_pool2d	✔	✔	✔	—	—
nn.functional.adaptive_max_pool3d	✔	✔	✔	—	—
nn.functional.alpha_dropout	✔	✔	✔	—	—
nn.functional.avg_pool1d	✔	✔	✔	—	✔
nn.functional.avg_pool2d	✔	✔	✔	—	✔
nn.functional.avg_pool3d	✔	—	—	—	✔
nn.functional.batch_norm	✔	✔	✔	—	—
nn.functional.bilinear	⚠ xfail	⚠ xfail	⚠ xfail	⚠ xfail	⚠ xfail
nn.functional.binary_cross_entropy	✔	✔	✔	—	—
nn.functional.binary_cross_entropy_with_logits	✔	✔	✔	—	—
nn.functional.celu	✔	✔	✔	—	—
nn.functional.channel_shuffle	✔	✔	✔	✔	✔
nn.functional.conv1d	✔	✔	✔	—	✔
nn.functional.conv2d	✔	✔	✔	—	✔
nn.functional.conv3d	✔	✔	✔	—	✔
nn.functional.conv_transpose1d	✔	✔	✔	—	✔
nn.functional.conv_transpose2d	✔	✔	✔	—	✔
nn.functional.conv_transpose3d	✔	✔	✔	—	✔
nn.functional.cosine_embedding_loss	✔	✔	✔	✔	✔
nn.functional.cosine_similarity	✔	✔	✔	—	—
nn.functional.cross_entropy	✔	✔	✔	—	—
nn.functional.ctc_loss	✔	—	—	—	—
nn.functional.dropout	✔	✔	✔	—	—
nn.functional.dropout2d	✔	✔	✔	—	—
nn.functional.dropout3d	✔	✔	✔	—	—
nn.functional.elu	✔	✔	✔	—	—
nn.functional.embedding	✔	✔	✔	—	—
nn.functional.embedding_bag	✔	✔	✔	—	—
nn.functional.fractional_max_pool2d	✔	✔	✔	—	—
nn.functional.fractional_max_pool3d	✔	✔	✔	—	—
nn.functional.gaussian_nll_loss	✔	✔	✔	—	—
nn.functional.gelu	✔	✔	✔	—	—
nn.functional.glu	✔	✔	✔	—	—
nn.functional.grid_sample	✔	✔	✔	—	—
nn.functional.group_norm	✔	✔	✔	—	—
nn.functional.hardshrink	✔	✔	✔	—	—
nn.functional.hardsigmoid	⚠ xfail	⚠ xfail	⚠ xfail	—	—
nn.functional.hardswish	✔	✔	✔	—	—
nn.functional.hardtanh	✔	✔	✔	✔	✔
nn.functional.hinge_embedding_loss	✔	✔	✔	—	—
nn.functional.huber_loss	✔	✔	✔	—	—
nn.functional.instance_norm	✔	✔	✔	—	—
nn.functional.kl_div	✔	✔	✔	—	—
nn.functional.l1_loss	✔	✔	✔	—	—
nn.functional.layer_norm	✔	✔	✔	—	—
nn.functional.leaky_relu	✔	✔	✔	—	—
nn.functional.linear	✔	✔	✔	⚠ xfail	⚠ xfail
nn.functional.local_response_norm	✔	✔	✔	—	✔
nn.functional.logsigmoid	✔	✔	✔	—	—
nn.functional.margin_ranking_loss	✔	✔	✔	✔	✔
nn.functional.max_pool1d	✔	✔	✔	—	—
nn.functional.max_pool2d	✔	✔	✔	✔	✔
nn.functional.max_pool3d	✔	✔	✔	✔	✔
nn.functional.max_unpool1d	✔	✔	✔	—	—
nn.functional.max_unpool2d	✔	✔	✔	—	—
nn.functional.max_unpool3d	✔	✔	✔	—	—
nn.functional.mish	✔	✔	✔	—	—
nn.functional.mse_loss	✔	✔	✔	—	—
nn.functional.multi_head_attention_forward	✔	✔	✔	—	—
nn.functional.multi_margin_loss	✘ no converter	—	—	—	—
nn.functional.multilabel_margin_loss	✘ no converter	—	—	—	—
nn.functional.multilabel_soft_margin_loss	✘ no converter	✘ no converter	✘ no converter	—	—
nn.functional.nll_loss	✔	✔	✔	—	—
nn.functional.normalize	✔	✔	✔	—	—
nn.functional.one_hot	—	—	—	—	✔
nn.functional.pairwise_distance	✔	✔	✔	✔	✔
nn.functional.pdist	✘ no converter	—	—	—	—
nn.functional.pixel_shuffle	✔	✔	✔	✔	✔
nn.functional.pixel_unshuffle	✔	✔	✔	✔	✔
nn.functional.poisson_nll_loss	✔	✔	✔	✔	✔
nn.functional.prelu	✔	✔	✔	—	—
nn.functional.relu	✔	✔	✔	⚠ xfail	⚠ xfail
nn.functional.relu6	✔	✔	✔	✔	✔
nn.functional.rms_norm	✔	✔	✔	—	—
nn.functional.rrelu	✔	—	✔	—	—
nn.functional.scaled_dot_product_attention	✔	✔	✔	—	—
nn.functional.selu	✔	✔	✔	—	—
nn.functional.silu	✔	✔	✔	—	—
nn.functional.smooth_l1_loss	✔	✔	✔	—	—
nn.functional.soft_margin_loss	✔	✔	✔	—	—
nn.functional.softmin	✔	✔	✔	—	—
nn.functional.softplus	✔	✔	✔	—	—
nn.functional.softshrink	✔	✔	✔	—	—
nn.functional.softsign	✔	✔	✔	⚠ xfail	⚠ xfail
nn.functional.tanhshrink	✔	✔	✔	⚠ xfail	⚠ xfail
nn.functional.threshold	✔	✔	✔	✔	✔
nn.functional.triplet_margin_loss	✔	✔	✔	✔	✔
nn.functional.triplet_margin_with_distance_loss	✔	✔	✔	✔	✔
nn.functional.unfold	✔	✔	✔	—	—
nn.functional.upsample_bilinear	✔	✔	✔	—	—
nn.functional.upsample_nearest	✔	✔	✔	—	—
nonzero	✔	✔	✔	✔	✔
nonzero_static	✔	✔	✔	✔	✔
norm	✔	✔	✔	—	—
ones	✔	✔	✔	✔	✔
ones_like	✔	✔	✔	✔	✔
ormqr	✔	—	—	—	—
outer	✔	✔	✔	✔	✔
pca_lowrank	✔	—	—	—	—
permute	✔	✔	✔	✔	✔
permute_copy	✔	✔	✔	✔	✔
pinverse	✘ no converter	—	—	—	—
polar	✔	—	—	—	—
positive	✘ no converter	✘ no converter	✘ no converter	✘ no converter	✘ no converter
pow	✔	✔	✔	✔	✔
prod	✔	✔	✔	⚠ xfail	✔
put	✔	✔	✔	✔	✔
qr	✘ no converter	—	—	—	—
quantile	✔	—	—	—	—
rad2deg	✘ no converter	✘ no converter	✘ no converter	✘ no converter	✘ no converter
randint_like	✔	✔	✔	✔	✔
ravel	✔	✔	✔	✔	✔
real	✘ no converter	✘ no converter	✘ no converter	✘ no converter	✘ no converter
reciprocal	✔	✔	✔	⚠ xfail	⚠ xfail
remainder	✔	✔	✔	✔	✔
renorm	✔	✔	✔	—	—
repeat	✔	✔	✔	✔	✔
repeat_interleave	✔	✔	✔	✔	✔
reshape	✔	✔	✔	✔	✔
reshape_as	✔	✔	✔	✔	✔
resize_	✔	✔	✔	✔	✔
resize_as_	✘ no converter	✘ no converter	✘ no converter	✘ no converter	✘ no converter
resolve_conj	✘ no converter	✘ no converter	✘ no converter	✘ no converter	✘ no converter
resolve_neg	✘ no converter	✘ no converter	✘ no converter	✘ no converter	✘ no converter
roll	✔	✔	✔	✔	✔
rot90	✔	✔	✔	✔	✔
round	✔	✔	✔	⚠ xfail	⚠ xfail
rsqrt	✔	✔	✔	⚠ xfail	⚠ xfail
rsub	✘ no converter	✘ no converter	✘ no converter	✘ no converter	✘ no converter
scalar_tensor	✔	✔	✔	✔	✔
scatter	✔	✔	✔	✔	✔
scatter_add	✔	✔	✔	✔	✔
searchsorted	✔	✔	✔	✔	✔
select	✔	✔	✔	✔	✔
select_scatter	✔	✔	✔	✔	✔
sgn	✘ no converter	✘ no converter	✘ no converter	✘ no converter	✘ no converter
short	✔	✔	✔	✔	✔
sigmoid	✔	✔	✔	⚠ xfail	⚠ xfail
sign	✔	✔	✔	✔	✔
signal.windows.bartlett	✔	—	—	—	—
signal.windows.blackman	✔	—	—	—	—
signal.windows.cosine	✔	—	—	—	—
signal.windows.exponential	✔	—	—	—	—
signal.windows.gaussian	✔	—	—	—	—
signal.windows.general_cosine	✔	—	—	—	—
signal.windows.general_hamming	✔	—	—	—	—
signal.windows.hamming	✔	—	—	—	—
signal.windows.hann	✔	—	—	—	—
signal.windows.kaiser	✔	—	—	—	—
signal.windows.nuttall	✔	—	—	—	—
signbit	✔	✔	✔	✔	✔
sin	✔	✔	✔	⚠ xfail	⚠ xfail
sinc	✘ no converter	✘ no converter	✘ no converter	✘ no converter	✘ no converter
sinh	✔	✔	✔	⚠ xfail	⚠ xfail
slice	✔	✔	✔	✔	✔
slice_scatter	✔	✔	✔	✔	✔
softmax	✔	✔	✔	—	—
sort	✘ no converter	✘ no converter	✘ no converter	✘ no converter	✘ no converter
sparse.sampled_addmm	✔	—	—	—	—
special.airy_ai	✘ no converter	—	—	✘ no converter	✘ no converter
special.bessel_j0	✘ no converter	—	—	✘ no converter	✘ no converter
special.bessel_j1	✘ no converter	—	—	✘ no converter	✘ no converter
special.bessel_y0	✘ no converter	—	—	✘ no converter	✘ no converter
special.bessel_y1	✘ no converter	—	—	✘ no converter	✘ no converter
special.chebyshev_polynomial_t	✘ no converter	—	—	✘ no converter	✘ no converter
special.chebyshev_polynomial_u	✘ no converter	—	—	✘ no converter	✘ no converter
special.chebyshev_polynomial_v	✘ no converter	—	—	✘ no converter	✘ no converter
special.chebyshev_polynomial_w	✘ no converter	—	—	✘ no converter	✘ no converter
special.entr	✘ no converter	✘ no converter	✘ no converter	✘ no converter	✘ no converter
special.erfcx	✘ no converter	—	—	✘ no converter	✘ no converter
special.hermite_polynomial_h	✘ no converter	—	—	✘ no converter	✘ no converter
special.hermite_polynomial_he	✘ no converter	—	—	✘ no converter	✘ no converter
special.i0e	✘ no converter	✘ no converter	✘ no converter	✘ no converter	✘ no converter
special.i1	✘ no converter	✘ no converter	✘ no converter	✘ no converter	✘ no converter
special.i1e	✘ no converter	✘ no converter	✘ no converter	✘ no converter	✘ no converter
special.laguerre_polynomial_l	✘ no converter	—	—	✘ no converter	✘ no converter
special.legendre_polynomial_p	✘ no converter	—	—	✘ no converter	✘ no converter
special.log_ndtr	✘ no converter	—	—	✘ no converter	✘ no converter
special.modified_bessel_i0	✘ no converter	—	—	✘ no converter	✘ no converter
special.modified_bessel_i1	✘ no converter	—	—	✘ no converter	✘ no converter
special.modified_bessel_k0	✘ no converter	—	—	✘ no converter	✘ no converter
special.modified_bessel_k1	✘ no converter	—	—	✘ no converter	✘ no converter
special.ndtr	✘ no converter	✘ no converter	✘ no converter	✘ no converter	✘ no converter
special.ndtri	✘ no converter	—	—	✘ no converter	✘ no converter
special.scaled_modified_bessel_k0	✘ no converter	—	—	✘ no converter	✘ no converter
special.scaled_modified_bessel_k1	✘ no converter	—	—	✘ no converter	✘ no converter
special.shifted_chebyshev_polynomial_t	✘ no converter	—	—	✘ no converter	✘ no converter
special.shifted_chebyshev_polynomial_u	✘ no converter	—	—	✘ no converter	✘ no converter
special.shifted_chebyshev_polynomial_v	✘ no converter	—	—	✘ no converter	✘ no converter
special.shifted_chebyshev_polynomial_w	✘ no converter	—	—	✘ no converter	✘ no converter
special.spherical_bessel_j0	✘ no converter	—	—	✘ no converter	✘ no converter
special.xlog1py	✘ no converter	✘ no converter	✘ no converter	✘ no converter	✘ no converter
special.zeta	✘ no converter	—	—	✘ no converter	✘ no converter
split	✔	✔	✔	✔	✔
split_with_sizes	✔	✔	✔	✔	✔
split_with_sizes_copy	✔	✔	✔	✔	✔
sqrt	✔	✔	✔	⚠ xfail	⚠ xfail
square	✔	✔	✔	✔	✔
squeeze	✔	✔	✔	✔	✔
squeeze_copy	✘ no converter	✘ no converter	✘ no converter	✘ no converter	✘ no converter
stack	✔	✔	✔	✔	✔
std	⚠ xfail	⚠ xfail	⚠ xfail	—	—
std_mean	⚠ xfail	⚠ xfail	⚠ xfail	—	—
stft	✔	—	—	—	—
sub	✔	✔	✔	✔	✔
sum	✔	✔	✔	⚠ xfail	✔
sum_to_size	✔	✔	✔	✔	✔
svd	✔	—	—	—	—
svd_lowrank	✔	—	—	—	—
t	✔	✔	✔	✔	✔
t_copy	✘ no converter	✘ no converter	✘ no converter	✘ no converter	✘ no converter
take	✔	✔	✔	✔	✔
take_along_dim	✔	✔	✔	✔	✔
tan	✔	✔	✔	⚠ xfail	⚠ xfail
tanh	✔	✔	✔	⚠ xfail	⚠ xfail
tensor_split	✔	✔	✔	✔	✔
tensordot	✔	✔	✔	✔	✔
tile	✔	✔	✔	✔	✔
to	✔	✔	✔	✔	✔
to_sparse	✘ no converter	✘ no converter	✘ no converter	✘ no converter	✘ no converter
topk	✔	✔	✔	✔	✔
torch._scaled_mm	—	—	—	—	—
torch._scaled_mm_v2	—	—	—	—	—
torch.ops.aten._efficient_attention_forward	—	—	—	—	—
torch.ops.aten._flash_attention_forward	—	—	—	—	—
torch.ops.aten._safe_softmax.default	✔	✔	✔	✔	✔
trace	✘ no converter	—	—	✘ no converter	✘ no converter
transpose	✔	✔	✔	✔	✔
transpose_copy	✔	✔	✔	✔	✔
trapezoid	✘ no converter	✘ no converter	✘ no converter	✘ no converter	✘ no converter
trapz	✘ no converter	✘ no converter	✘ no converter	✘ no converter	✘ no converter
triangular_solve	✘ no converter	—	—	—	—
tril	✔	✔	✔	⚠ xfail	✔
tril_indices	—	—	—	✔	✔
triu	✔	✔	✔	⚠ xfail	✔
triu_indices	—	—	—	✔	✔
true_divide	✔	✔	✔	✔	✔
trunc	✔	✔	✔	⚠ xfail	⚠ xfail
unbind	✔	✔	✔	✔	✔
unbind_copy	✔	✔	✔	✔	✔
unflatten	✔	✔	✔	✔	✔
unfold	✔	✔	✔	✔	✔
unfold_copy	✔	✔	✔	✔	✔
unique	✔	✔	✔	✔	✔
unique_consecutive	✔	✔	✔	✔	✔
unravel_index	—	—	—	✔	✔
unsafe_chunk	✔	✔	✔	✔	✔
unsafe_split	✔	✔	✔	✔	✔
unsqueeze	✔	✔	✔	✔	✔
unsqueeze_copy	✔	✔	✔	✔	✔
var	✘ no converter	✘ no converter	✘ no converter	—	—
var_mean	✘ no converter	✘ no converter	✘ no converter	—	—
vdot	✘ no converter	✘ no converter	✘ no converter	✘ no converter	✘ no converter
view	✔	✔	✔	✔	✔
view_as	✘ no converter	✘ no converter	✘ no converter	✘ no converter	✘ no converter
view_as_complex	✔	✔	—	—	—
view_as_real	—	—	—	—	—
view_copy	✔	✔	✔	✔	✔
vsplit	✔	✔	✔	✔	✔
vstack	✔	✔	✔	✔	✔
where	✔	✔	✔	✔	✔
xlogy	✔	✔	✔	⚠ xfail	⚠ xfail
zero_	✘ no converter	✘ no converter	✘ no converter	✘ no converter	✘ no converter
zeros	✔	✔	✔	✔	✔
zeros_like	✔	✔	✔	✔	✔

[runpythonerror] W0429 17:13:15.383000 19108 torch/_native/cutedsl_utils.py:55] CuTeDSL operators require optional Python packages nvidia-cutlass-dsl and apache-tvm-ffi; missing optional dependency nvidia_cutlass_dsl (importlib.util.find_spec(nvidia_cutlass_dsl) failed)

New-tracing export path

New-tracing export path (tracing=TracingMode.NEW_TRACING)

Op	float32	float16	bfloat16	int32	int64
H	✘ no converter	✘ no converter	✘ no converter	✘ no converter	✘ no converter
T	✔	✔	✔	✔	✔
__getitem__	✔	✔	✔	✔	✔
__radd__	✔	✔	✔	✔	✔
__rand__	—	—	—	✔	✔
__rdiv__	✔	✔	✔	⚠ xfail	⚠ xfail
__rmatmul__	✔	✔	✔	✔	✔
__rmod__	✔	✔	✔	—	—
__rmul__	✔	✔	✔	✔	✔
__ror__	—	—	—	✔	✔
__rpow__	✔	✔	✔	⚠ xfail	⚠ xfail
__rsub__	✘ no converter	✘ no converter	✘ no converter	✘ no converter	✘ no converter
__rxor__	—	—	—	⚠ xfail	⚠ xfail
_batch_norm_with_update	✔	✔	✔	—	—
_chunk_cat	✔	✔	✔	✔	✔
_native_batch_norm_legit	✔	✔	✔	—	—
_softmax_backward_data	✔	✔	✔	—	—
_unsafe_masked_index	✔	✔	✔	✔	✔
_unsafe_masked_index_put_accumulate	✔	✔	✔	✔	✔
_upsample_bilinear2d_aa	✔	—	—	—	—
abs	✔	✔	✔	✔	✔
acos	✔	✔	✔	⚠ xfail	⚠ xfail
acosh	✔	✔	✔	⚠ xfail	⚠ xfail
add	✔	✔	✔	✔	✔
addbmm	✔	✔	✔	✔	✔
addcdiv	✘ no converter	✘ no converter	✘ no converter	—	—
addcmul	✔	⚠ xfail	⚠ xfail	✔	✔
addmm	✔	✔	✔	✔	✔
addmv	✔	✔	✔	✔	✔
addr	✘ no converter	✘ no converter	✘ no converter	✘ no converter	✘ no converter
alias_copy	✘ no converter	✘ no converter	✘ no converter	✘ no converter	✘ no converter
all	✔	✔	✔	✔	✔
allclose	✔	✔	✔	—	—
amax	✔	✔	✔	✔	✔
amin	✔	✔	✔	✔	✔
aminmax	✔	✔	✔	✔	✔
angle	✔	✔	✔	⚠ xfail	⚠ xfail
any	✔	✔	✔	✔	✔
arange	✔	✔	✔	✔	✔
argmax	✔	✔	✔	✔	✔
argmin	✔	✔	✔	✔	✔
argsort	✔	⚠ xfail	⚠ xfail	✔	✔
argwhere	✔	✔	✔	✔	✔
as_strided	✔	✔	✔	✔	✔
as_strided_copy	✔	✔	✔	✔	✔
as_strided_scatter	✔	✔	✔	✔	✔
asin	✔	✔	✔	⚠ xfail	⚠ xfail
asinh	✔	✔	✔	⚠ xfail	⚠ xfail
atan	✔	✔	✔	⚠ xfail	⚠ xfail
atan2	✔	✔	✔	⚠ xfail	⚠ xfail
atanh	✔	✔	✔	⚠ xfail	⚠ xfail
atleast_1d	✔	✔	✔	✔	✔
atleast_2d	✔	✔	✔	✔	✔
atleast_3d	✔	✔	✔	✔	✔
baddbmm	✔	✔	✔	✔	✔
bernoulli	✘ no converter	✘ no converter	✘ no converter	—	—
bfloat16	⚠ xfail	⚠ xfail	⚠ xfail	⚠ xfail	⚠ xfail
bincount	—	—	—	✔	✔
bitwise_and	—	—	—	✔	✔
bitwise_left_shift	—	—	—	⚠ xfail	⚠ xfail
bitwise_not	—	—	—	✔	✔
bitwise_or	—	—	—	✔	✔
bitwise_right_shift	—	—	—	⚠ xfail	⚠ xfail
bitwise_xor	—	—	—	⚠ xfail	⚠ xfail
block_diag	✔	✔	✔	✔	✔
bmm	✔	✔	⚠ xfail	✔	✔
bool	✔	✔	✔	✔	✔
broadcast_shapes	✔	—	—	—	—
broadcast_tensors	⚠ xfail	⚠ xfail	⚠ xfail	⚠ xfail	⚠ xfail
broadcast_to	✔	✔	✔	✔	✔
bucketize	✔	✔	✔	✔	✔
byte	✔	✔	✔	✔	✔
cartesian_prod	✔	✔	✔	✔	✔
cat	✔	✔	✔	✔	✔
cdist	✔	—	—	—	—
cdouble	✔	✔	✔	✔	✔
ceil	✔	✔	✔	⚠ xfail	⚠ xfail
cfloat	✔	✔	✔	✔	✔
chalf	✔	✔	✔	✔	✔
char	✔	✔	✔	✔	✔
cholesky	✔	—	—	—	—
cholesky_inverse	⚠ xfail	—	—	—	—
cholesky_solve	⚠ xfail	—	—	—	—
chunk	✔	✔	✔	✔	✔
clamp	⚠ xfail	⚠ xfail	⚠ xfail	⚠ xfail	⚠ xfail
clamp_max	✔	✔	✔	✔	✔
clamp_min	✔	✔	✔	✔	✔
clone	✔	✔	✔	✔	✔
column_stack	✔	✔	✔	✔	✔
combinations	✔	✔	✔	✔	✔
complex	✔	✔	—	—	—
conj	✔	✔	✔	✔	✔
conj_physical	✘ no converter	✘ no converter	✘ no converter	✘ no converter	✘ no converter
constant_pad_nd	✔	✔	✔	✔	✔
contiguous	✔	✔	✔	✔	✔
copysign	✘ no converter	✘ no converter	✘ no converter	✘ no converter	✘ no converter
corrcoef	⚠ xfail	⚠ xfail	⚠ xfail	⚠ xfail	⚠ xfail
cos	✔	✔	✔	⚠ xfail	⚠ xfail
cosh	✔	✔	✔	⚠ xfail	⚠ xfail
count_nonzero	✔	✔	✔	✔	✔
cov	⚠ xfail	⚠ xfail	⚠ xfail	⚠ xfail	⚠ xfail
cross	✘ no converter	✘ no converter	✘ no converter	✘ no converter	✘ no converter
cummax	✔	✔	✔	✔	✔
cummin	✔	✔	✔	✔	✔
cumprod	✔	✔	✔	✔	✔
cumsum	✔	✔	✔	✔	✔
cumulative_trapezoid	✘ no converter	✘ no converter	✘ no converter	✘ no converter	✘ no converter
deg2rad	✘ no converter	✘ no converter	✘ no converter	✘ no converter	✘ no converter
diag	✔	✔	✔	✔	✔
diag_embed	✔	✔	✔	✔	✔
diagflat	✘ no converter	✘ no converter	✘ no converter	✘ no converter	✘ no converter
diagonal	✘ no converter	✘ no converter	✘ no converter	✘ no converter	✘ no converter
diagonal_copy	✘ no converter	✘ no converter	✘ no converter	✘ no converter	✘ no converter
diagonal_scatter	✘ no converter	✘ no converter	✘ no converter	✘ no converter	✘ no converter
diff	✔	✔	✔	✔	✔
digamma	✘ no converter	✘ no converter	✘ no converter	✘ no converter	✘ no converter
dist	✔	✔	✔	—	—
dot	✘ no converter	✘ no converter	✘ no converter	✘ no converter	✘ no converter
double	✔	✔	✔	✔	✔
dsplit	✔	✔	✔	✔	✔
dstack	✔	✔	✔	✔	✔
einsum	✔	✔	✔	✔	✔
empty_permuted	✔	✔	✔	✔	✔
empty_strided	✔	✔	✔	✔	✔
eq	✔	✔	✔	✔	✔
equal	✔	✔	✔	✔	✔
erf	✔	✔	✔	⚠ xfail	⚠ xfail
erfc	✔	✔	✔	⚠ xfail	⚠ xfail
erfinv	✔	✔	⚠ xfail	⚠ xfail	⚠ xfail
exp	✔	✔	✔	⚠ xfail	⚠ xfail
exp2	✘ no converter	✘ no converter	✘ no converter	✘ no converter	✘ no converter
expand	✔	✔	✔	✔	✔
expand_as	✔	✔	✔	✔	✔
expand_copy	✔	✔	✔	✔	✔
expm1	✔	⚠ xfail	⚠ xfail	⚠ xfail	⚠ xfail
eye	✔	✔	✔	✔	✔
fft.fft	✔	—	—	✔	✔
fft.fft2	✔	—	—	✔	✔
fft.fftn	✔	—	—	✔	✔
fft.fftshift	✔	✔	✔	✔	✔
fft.hfft	✔	—	—	✔	✔
fft.hfft2	✔	—	—	✔	✔
fft.hfftn	✔	—	—	✔	✔
fft.ifft	✔	—	—	✔	✔
fft.ifft2	✔	—	—	✔	✔
fft.ifftn	✔	—	—	✔	✔
fft.ifftshift	✔	✔	✔	✔	✔
fft.ihfft	✔	—	—	✔	✔
fft.ihfft2	✔	—	—	✔	✔
fft.ihfftn	✔	—	—	✔	✔
fft.irfft	✔	—	—	✔	✔
fft.irfft2	✔	—	—	✔	✔
fft.irfftn	✔	—	—	✔	✔
fft.rfft	✔	—	—	✔	✔
fft.rfft2	✔	—	—	✔	✔
fft.rfftn	✔	—	—	✔	✔
fill	✔	✔	✔	✔	✔
flatten	✔	✔	✔	✔	✔
flip	✔	✔	✔	✔	✔
fliplr	✔	✔	✔	✔	✔
flipud	✔	✔	✔	✔	✔
float	✔	✔	✔	✔	✔
float_power	✔	✔	✔	✔	✔
floor	✔	✔	✔	⚠ xfail	⚠ xfail
floor_divide	✔	✔	✔	⚠ xfail	⚠ xfail
fmax	✔	✔	✔	✔	✔
fmin	✔	✔	✔	✔	✔
fmod	✔	✔	✔	✔	✔
frac	✔	✔	✔	—	—
frexp	✔	✔	⚠ xfail	—	—
full	✔	✔	✔	✔	✔
full_like	✔	✔	✔	✔	✔
gather	✔	✔	✔	✔	✔
gcd	—	—	—	⚠ xfail	⚠ xfail
ge	✔	✔	✔	✔	✔
geqrf	✔	—	—	—	—
gradient	✔	✔	✔	✔	✔
grid_sampler_2d	✔	✔	✔	—	—
grid_sampler_3d	✔	✔	✔	—	—
gt	✔	✔	✔	✔	✔
half	✔	✔	✔	✔	✔
hash_tensor	✘ no converter	✘ no converter	✘ no converter	✘ no converter	✘ no converter
heaviside	✔	✔	✔	✔	✔
histc	✔	✔	✔	—	—
histogram	✔	—	—	—	—
histogramdd	✔	—	—	—	—
hsplit	✔	✔	✔	✔	✔
hstack	✔	✔	✔	✔	✔
hypot	✘ no converter	✘ no converter	✘ no converter	—	—
i0	✘ no converter	✘ no converter	✘ no converter	✘ no converter	✘ no converter
igamma	✘ no converter	✘ no converter	✘ no converter	—	—
igammac	✘ no converter	✘ no converter	✘ no converter	—	—
imag	—	—	—	—	—
index_add	✔	✔	✔	✔	✔
index_copy	✔	✔	✔	✔	✔
index_fill	✔	✔	✔	✔	✔
index_put	✔	✔	✔	✔	✔
index_select	✔	✔	✔	✔	✔
inner	✘ no converter	✘ no converter	✘ no converter	✘ no converter	✘ no converter
int	✔	✔	✔	✔	✔
isclose	✔	✔	✔	✔	✔
isfinite	✔	✔	✔	✔	✔
isin	✔	✔	✔	✔	✔
isinf	✔	✔	✔	⚠ xfail	⚠ xfail
isnan	✔	✔	✔	⚠ xfail	⚠ xfail
isneginf	✔	✔	✔	✔	✔
isposinf	✔	✔	✔	✔	✔
isreal	✘ no converter	✘ no converter	✘ no converter	✘ no converter	✘ no converter
istft	—	—	—	—	—
item	✔	✔	✔	✔	✔
jiterator_2inputs_2outputs	⚠ xfail	⚠ xfail	⚠ xfail	⚠ xfail	⚠ xfail
jiterator_4inputs_with_extra_args	⚠ xfail	⚠ xfail	⚠ xfail	⚠ xfail	⚠ xfail
jiterator_binary	⚠ xfail	⚠ xfail	⚠ xfail	⚠ xfail	⚠ xfail
jiterator_binary_return_by_ref	⚠ xfail	⚠ xfail	⚠ xfail	⚠ xfail	⚠ xfail
jiterator_unary	⚠ xfail	⚠ xfail	⚠ xfail	⚠ xfail	⚠ xfail
kron	✘ no converter	✘ no converter	✘ no converter	✘ no converter	✘ no converter
kthvalue	✔	✔	✔	✔	✔
lcm	—	—	—	⚠ xfail	⚠ xfail
ldexp	✘ no converter	✘ no converter	✘ no converter	✘ no converter	✘ no converter
le	✔	✔	✔	✔	✔
lerp	✔	✔	✔	—	—
lgamma	✘ no converter	✘ no converter	✘ no converter	✘ no converter	✘ no converter
linalg.cholesky	✔	—	—	—	—
linalg.cholesky_ex	✔	—	—	—	—
linalg.cond	✘ no converter	—	—	—	—
linalg.cross	✔	✔	✔	✔	✔
linalg.det	✔	—	—	—	—
linalg.diagonal	✘ no converter	✘ no converter	✘ no converter	✘ no converter	✘ no converter
linalg.eig	✔	—	—	—	—
linalg.eigh	✔	—	—	—	—
linalg.eigvals	✔	—	—	—	—
linalg.eigvalsh	✔	—	—	—	—
linalg.householder_product	✘ no converter	—	—	—	—
linalg.inv	✘ no converter	—	—	—	—
linalg.inv_ex	✘ no converter	—	—	—	—
linalg.ldl_factor	✔	—	—	—	—
linalg.ldl_factor_ex	✔	—	—	—	—
linalg.ldl_solve	✔	—	—	—	—
linalg.lstsq	✔	—	—	—	—
linalg.lu	✔	—	—	—	—
linalg.lu_factor	✔	—	—	—	—
linalg.lu_factor_ex	✔	—	—	—	—
linalg.lu_solve	✔	—	—	—	—
linalg.matrix_norm	✔	✔	✔	—	—
linalg.matrix_power	✔	—	—	—	—
linalg.matrix_rank	✔	—	—	—	—
linalg.multi_dot	✔	✔	✔	✔	✔
linalg.norm	✔	✔	✔	—	—
linalg.pinv	✔	—	—	—	—
linalg.qr	✘ no converter	—	—	—	—
linalg.slogdet	✔	—	—	—	—
linalg.solve	✘ no converter	—	—	—	—
linalg.solve_ex	✘ no converter	—	—	—	—
linalg.solve_triangular	✔	—	—	—	—
linalg.svd	✔	—	—	—	—
linalg.svdvals	✘ no converter	—	—	—	—
linalg.tensorinv	✔	—	—	—	—
linalg.tensorsolve	✔	—	—	—	—
linalg.vander	✔	—	—	✔	✔
linalg.vecdot	✔	✔	✔	—	—
linalg.vector_norm	✔	✔	✔	—	—
linspace	✔	✔	✔	✔	✔
log	✔	✔	✔	⚠ xfail	⚠ xfail
log10	✔	✔	⚠ xfail	⚠ xfail	⚠ xfail
log1p	✔	✔	⚠ xfail	⚠ xfail	⚠ xfail
log2	✔	✔	⚠ xfail	⚠ xfail	⚠ xfail
log_softmax	✔	✔	✔	—	—
logaddexp	✔	✔	⚠ xfail	—	—
logaddexp2	✔	✔	⚠ xfail	—	—
logcumsumexp	✔	✔	✔	—	—
logdet	✔	—	—	—	—
logical_and	✔	✔	✔	✔	✔
logical_not	⚠ xfail	⚠ xfail	⚠ xfail	⚠ xfail	⚠ xfail
logical_or	✔	✔	✔	✔	✔
logical_xor	✔	✔	✔	✔	✔
logit	✔	✔	✔	⚠ xfail	⚠ xfail
logspace	✔	✔	✔	✔	✔
logsumexp	✔	✔	✔	✔	✔
long	✔	✔	✔	✔	✔
lt	✔	✔	✔	✔	✔
lu	✔	—	—	—	—
lu_solve	✘ no converter	—	—	—	—
lu_unpack	✘ no converter	—	—	—	—
mH	✔	✔	✔	✔	✔
mT	✔	✔	✔	✔	✔
masked.amax	✔	✔	✔	✔	✔
masked.amin	✔	✔	✔	✔	✔
masked.argmax	✔	✔	✔	✔	✔
masked.argmin	✔	✔	✔	✔	✔
masked.cumprod	✔	✔	✔	✔	✔
masked.cumsum	✔	✔	✔	✔	✔
masked.log_softmax	✔	✔	✔	—	—
masked.logaddexp	✔	✔	✔	—	—
masked.logsumexp	✔	✔	✔	✔	✔
masked.mean	✔	✔	✔	—	—
masked.median	✔	✔	✔	—	—
masked.norm	✔	✔	✔	—	—
masked.normalize	✔	✔	✔	—	—
masked.prod	✔	✔	✔	✔	✔
masked.softmax	✔	✔	✔	—	—
masked.softmin	✔	✔	✔	—	—
masked.std	✔	✔	✔	✔	✔
masked.sum	✔	✔	✔	✔	✔
masked.var	✔	✔	✔	✔	✔
masked_fill	✔	✔	✔	✔	✔
masked_scatter	✔	✔	✔	✔	✔
masked_select	✘ no converter	✘ no converter	✘ no converter	✘ no converter	✘ no converter
matmul	✔	✔	✔	✔	✔
matrix_exp	✘ no converter	✘ no converter	✘ no converter	—	—
max_pool2d_with_indices_backward	✔	✔	✔	—	—
maximum	✔	✔	✔	✔	✔
mean	✔	✔	✔	—	—
median	✘ no converter	✘ no converter	✘ no converter	✘ no converter	✘ no converter
minimum	✔	✔	✔	✔	✔
mm	✔	✔	✔	✔	✔
mode	✘ no converter	✘ no converter	✘ no converter	✘ no converter	✘ no converter
movedim	✔	✔	✔	✔	✔
msort	✘ no converter	✘ no converter	✘ no converter	✘ no converter	✘ no converter
mul	✔	✔	✔	✔	✔
multinomial	✔	✔	✔	—	—
mv	✘ no converter	✘ no converter	✘ no converter	✘ no converter	✘ no converter
nan_to_num	⚠ xfail	⚠ xfail	⚠ xfail	⚠ xfail	⚠ xfail
nanmean	✔	✔	✔	—	—
nanmedian	✘ no converter	✘ no converter	✘ no converter	✘ no converter	✘ no converter
nanquantile	✔	—	—	—	—
nansum	✔	✔	✔	⚠ xfail	⚠ xfail
narrow	✔	✔	✔	✔	✔
narrow_copy	✔	✔	✔	✔	✔
native_batch_norm	✔	✔	✔	—	—
native_dropout_backward	✔	✔	✔	✔	✔
native_layer_norm	✔	✔	✔	—	—
ne	✔	✔	✔	✔	✔
neg	✔	✔	✔	✔	✔
new_empty	✔	✔	✔	✔	✔
new_empty_strided	✔	✔	✔	✔	✔
new_full	✔	✔	✔	✔	✔
new_ones	✔	✔	✔	✔	✔
new_zeros	✔	✔	✔	✔	✔
nextafter	✘ no converter	✘ no converter	✘ no converter	—	—
nn.functional.adaptive_avg_pool1d	✔	✔	✔	—	—
nn.functional.adaptive_avg_pool2d	✔	✔	✔	—	—
nn.functional.adaptive_avg_pool3d	✔	✔	✔	—	—
nn.functional.adaptive_max_pool1d	✔	✔	✔	—	—
nn.functional.adaptive_max_pool2d	✔	✔	✔	—	—
nn.functional.adaptive_max_pool3d	✔	✔	✔	—	—
nn.functional.alpha_dropout	✔	✔	✔	—	—
nn.functional.avg_pool1d	✔	✔	✔	—	✔
nn.functional.avg_pool2d	✔	✔	✔	—	✔
nn.functional.avg_pool3d	✔	—	—	—	✔
nn.functional.batch_norm	✔	✔	✔	—	—
nn.functional.bilinear	✔	✔	✔	✔	✔
nn.functional.binary_cross_entropy	✔	✔	✔	—	—
nn.functional.binary_cross_entropy_with_logits	✔	✔	✔	—	—
nn.functional.celu	✔	✔	✔	—	—
nn.functional.channel_shuffle	✔	✔	✔	✔	✔
nn.functional.conv1d	✔	✔	✔	—	✔
nn.functional.conv2d	✔	✔	✔	—	✔
nn.functional.conv3d	✔	✔	✔	—	✔
nn.functional.conv_transpose1d	✔	✔	✔	—	✔
nn.functional.conv_transpose2d	✔	✔	✔	—	✔
nn.functional.conv_transpose3d	✔	✔	✔	—	✔
nn.functional.cosine_embedding_loss	✔	✔	✔	✔	✔
nn.functional.cosine_similarity	✔	✔	✔	—	—
nn.functional.cross_entropy	✔	✔	✔	—	—
nn.functional.ctc_loss	✔	—	—	—	—
nn.functional.dropout	✔	✔	✔	—	—
nn.functional.dropout2d	✔	✔	✔	—	—
nn.functional.dropout3d	✔	✔	✔	—	—
nn.functional.elu	✔	✔	✔	—	—
nn.functional.embedding	✔	✔	✔	—	—
nn.functional.embedding_bag	✔	✔	✔	—	—
nn.functional.fractional_max_pool2d	✔	✔	✔	—	—
nn.functional.fractional_max_pool3d	✔	✔	✔	—	—
nn.functional.gaussian_nll_loss	✔	✔	✔	—	—
nn.functional.gelu	✔	✔	✔	—	—
nn.functional.glu	✔	✔	✔	—	—
nn.functional.grid_sample	✔	✔	✔	—	—
nn.functional.group_norm	✔	✔	✔	—	—
nn.functional.hardshrink	✔	✔	✔	—	—
nn.functional.hardsigmoid	✔	✔	✔	—	—
nn.functional.hardswish	✔	✔	✔	—	—
nn.functional.hardtanh	✔	✔	✔	✔	✔
nn.functional.hinge_embedding_loss	✔	✔	✔	—	—
nn.functional.huber_loss	✔	✔	✔	—	—
nn.functional.instance_norm	✔	✔	✔	—	—
nn.functional.kl_div	✔	✔	✔	—	—
nn.functional.l1_loss	✔	✔	✔	—	—
nn.functional.layer_norm	✔	✔	✔	—	—
nn.functional.leaky_relu	✔	✔	✔	—	—
nn.functional.linear	✔	✔	✔	⚠ xfail	⚠ xfail
nn.functional.local_response_norm	✔	✔	✔	—	✔
nn.functional.logsigmoid	✔	✔	✔	—	—
nn.functional.margin_ranking_loss	✔	✔	✔	✔	✔
nn.functional.max_pool1d	✔	✔	✔	—	—
nn.functional.max_pool2d	✔	✔	✔	✔	✔
nn.functional.max_pool3d	✔	✔	✔	✔	✔
nn.functional.max_unpool1d	✔	✔	✔	—	—
nn.functional.max_unpool2d	✔	✔	✔	—	—
nn.functional.max_unpool3d	✔	✔	✔	—	—
nn.functional.mish	✔	✔	✔	—	—
nn.functional.mse_loss	✔	✔	✔	—	—
nn.functional.multi_head_attention_forward	✔	✔	✔	—	—
nn.functional.multi_margin_loss	✘ no converter	—	—	—	—
nn.functional.multilabel_margin_loss	✘ no converter	—	—	—	—
nn.functional.multilabel_soft_margin_loss	✘ no converter	✘ no converter	✘ no converter	—	—
nn.functional.nll_loss	✔	✔	✔	—	—
nn.functional.normalize	✔	✔	✔	—	—
nn.functional.one_hot	—	—	—	—	✔
nn.functional.pairwise_distance	✔	✔	✔	✔	✔
nn.functional.pdist	✘ no converter	—	—	—	—
nn.functional.pixel_shuffle	✔	✔	✔	✔	✔
nn.functional.pixel_unshuffle	✔	✔	✔	✔	✔
nn.functional.poisson_nll_loss	✔	✔	✔	✔	✔
nn.functional.prelu	✔	✔	✔	—	—
nn.functional.relu	✔	✔	✔	⚠ xfail	⚠ xfail
nn.functional.relu6	✔	✔	✔	✔	✔
nn.functional.rms_norm	✔	✔	✔	—	—
nn.functional.rrelu	✔	—	✔	—	—
nn.functional.scaled_dot_product_attention	✔	✔	✔	—	—
nn.functional.selu	✔	✔	✔	—	—
nn.functional.silu	✔	✔	✔	—	—
nn.functional.smooth_l1_loss	✔	✔	✔	—	—
nn.functional.soft_margin_loss	✔	✔	✔	—	—
nn.functional.softmin	✔	✔	✔	—	—
nn.functional.softplus	✔	✔	✔	—	—
nn.functional.softshrink	✔	✔	✔	—	—
nn.functional.softsign	✔	✔	✔	⚠ xfail	⚠ xfail
nn.functional.tanhshrink	✔	✔	✔	⚠ xfail	⚠ xfail
nn.functional.threshold	✔	✔	✔	✔	✔
nn.functional.triplet_margin_loss	✔	✔	✔	✔	✔
nn.functional.triplet_margin_with_distance_loss	✔	✔	✔	✔	✔
nn.functional.unfold	✔	✔	✔	—	—
nn.functional.upsample_bilinear	✔	✔	✔	—	—
nn.functional.upsample_nearest	✔	✔	✔	—	—
nonzero	✔	✔	✔	✔	✔
nonzero_static	✔	✔	✔	✔	✔
norm	✔	✔	✔	—	—
ones	✔	✔	✔	✔	✔
ones_like	✔	✔	✔	✔	✔
ormqr	✔	—	—	—	—
outer	✔	✔	✔	✔	✔
pca_lowrank	✔	—	—	—	—
permute	✔	✔	✔	✔	✔
permute_copy	✔	✔	✔	✔	✔
pinverse	✘ no converter	—	—	—	—
polar	✔	—	—	—	—
positive	✘ no converter	✘ no converter	✘ no converter	✘ no converter	✘ no converter
pow	✔	✔	✔	✔	✔
prod	✔	✔	✔	⚠ xfail	✔
put	✔	✔	✔	✔	✔
qr	✘ no converter	—	—	—	—
quantile	✔	—	—	—	—
rad2deg	✘ no converter	✘ no converter	✘ no converter	✘ no converter	✘ no converter
randint_like	✔	✔	✔	✔	✔
ravel	✔	✔	✔	✔	✔
real	✘ no converter	✘ no converter	✘ no converter	✘ no converter	✘ no converter
reciprocal	✔	✔	✔	⚠ xfail	⚠ xfail
remainder	✔	✔	✔	✔	✔
renorm	✔	✔	✔	—	—
repeat	✔	✔	✔	✔	✔
repeat_interleave	✔	✔	✔	✔	✔
reshape	✔	✔	✔	✔	✔
reshape_as	✔	✔	✔	✔	✔
resize_	✔	✔	✔	✔	✔
resize_as_	✘ no converter	✘ no converter	✘ no converter	✘ no converter	✘ no converter
resolve_conj	✘ no converter	✘ no converter	✘ no converter	✘ no converter	✘ no converter
resolve_neg	✘ no converter	✘ no converter	✘ no converter	✘ no converter	✘ no converter
roll	✔	✔	✔	✔	✔
rot90	✔	✔	✔	✔	✔
round	✔	✔	✔	⚠ xfail	⚠ xfail
rsqrt	✔	✔	✔	⚠ xfail	⚠ xfail
rsub	✘ no converter	✘ no converter	✘ no converter	✘ no converter	✘ no converter
scalar_tensor	✔	✔	✔	✔	✔
scatter	✔	✔	✔	✔	✔
scatter_add	✔	✔	✔	✔	✔
searchsorted	✔	✔	✔	✔	✔
select	✔	✔	✔	✔	✔
select_scatter	✔	✔	✔	✔	✔
sgn	✘ no converter	✘ no converter	✘ no converter	✘ no converter	✘ no converter
short	✔	✔	✔	✔	✔
sigmoid	✔	✔	✔	⚠ xfail	⚠ xfail
sign	✔	✔	✔	✔	✔
signal.windows.bartlett	✔	—	—	—	—
signal.windows.blackman	✔	—	—	—	—
signal.windows.cosine	✔	—	—	—	—
signal.windows.exponential	✔	—	—	—	—
signal.windows.gaussian	✔	—	—	—	—
signal.windows.general_cosine	✔	—	—	—	—
signal.windows.general_hamming	✔	—	—	—	—
signal.windows.hamming	✔	—	—	—	—
signal.windows.hann	✔	—	—	—	—
signal.windows.kaiser	✔	—	—	—	—
signal.windows.nuttall	✔	—	—	—	—
signbit	✔	✔	✔	✔	✔
sin	✔	✔	✔	⚠ xfail	⚠ xfail
sinc	✘ no converter	✘ no converter	✘ no converter	✘ no converter	✘ no converter
sinh	✔	✔	✔	⚠ xfail	⚠ xfail
slice	✔	✔	✔	✔	✔
slice_scatter	✔	✔	✔	✔	✔
softmax	✔	✔	✔	—	—
sort	✘ no converter	✘ no converter	✘ no converter	✘ no converter	✘ no converter
sparse.sampled_addmm	✔	—	—	—	—
special.airy_ai	✘ no converter	—	—	✘ no converter	✘ no converter
special.bessel_j0	✘ no converter	—	—	✘ no converter	✘ no converter
special.bessel_j1	✘ no converter	—	—	✘ no converter	✘ no converter
special.bessel_y0	✘ no converter	—	—	✘ no converter	✘ no converter
special.bessel_y1	✘ no converter	—	—	✘ no converter	✘ no converter
special.chebyshev_polynomial_t	✘ no converter	—	—	✘ no converter	✘ no converter
special.chebyshev_polynomial_u	✘ no converter	—	—	✘ no converter	✘ no converter
special.chebyshev_polynomial_v	✘ no converter	—	—	✘ no converter	✘ no converter
special.chebyshev_polynomial_w	✘ no converter	—	—	✘ no converter	✘ no converter
special.entr	✘ no converter	✘ no converter	✘ no converter	✘ no converter	✘ no converter
special.erfcx	✘ no converter	—	—	✘ no converter	✘ no converter
special.hermite_polynomial_h	✘ no converter	—	—	✘ no converter	✘ no converter
special.hermite_polynomial_he	✘ no converter	—	—	✘ no converter	✘ no converter
special.i0e	✘ no converter	✘ no converter	✘ no converter	✘ no converter	✘ no converter
special.i1	✘ no converter	✘ no converter	✘ no converter	✘ no converter	✘ no converter
special.i1e	✘ no converter	✘ no converter	✘ no converter	✘ no converter	✘ no converter
special.laguerre_polynomial_l	✘ no converter	—	—	✘ no converter	✘ no converter
special.legendre_polynomial_p	✘ no converter	—	—	✘ no converter	✘ no converter
special.log_ndtr	✘ no converter	—	—	✘ no converter	✘ no converter
special.modified_bessel_i0	✘ no converter	—	—	✘ no converter	✘ no converter
special.modified_bessel_i1	✘ no converter	—	—	✘ no converter	✘ no converter
special.modified_bessel_k0	✘ no converter	—	—	✘ no converter	✘ no converter
special.modified_bessel_k1	✘ no converter	—	—	✘ no converter	✘ no converter
special.ndtr	✘ no converter	✘ no converter	✘ no converter	✘ no converter	✘ no converter
special.ndtri	✘ no converter	—	—	✘ no converter	✘ no converter
special.scaled_modified_bessel_k0	✘ no converter	—	—	✘ no converter	✘ no converter
special.scaled_modified_bessel_k1	✘ no converter	—	—	✘ no converter	✘ no converter
special.shifted_chebyshev_polynomial_t	✘ no converter	—	—	✘ no converter	✘ no converter
special.shifted_chebyshev_polynomial_u	✘ no converter	—	—	✘ no converter	✘ no converter
special.shifted_chebyshev_polynomial_v	✘ no converter	—	—	✘ no converter	✘ no converter
special.shifted_chebyshev_polynomial_w	✘ no converter	—	—	✘ no converter	✘ no converter
special.spherical_bessel_j0	✘ no converter	—	—	✘ no converter	✘ no converter
special.xlog1py	✘ no converter	✘ no converter	✘ no converter	✘ no converter	✘ no converter
special.zeta	✘ no converter	—	—	✘ no converter	✘ no converter
split	✔	✔	✔	✔	✔
split_with_sizes	✔	✔	✔	✔	✔
split_with_sizes_copy	✔	✔	✔	✔	✔
sqrt	✔	✔	✔	⚠ xfail	⚠ xfail
square	✔	✔	✔	✔	✔
squeeze	✔	✔	✔	✔	✔
squeeze_copy	✘ no converter	✘ no converter	✘ no converter	✘ no converter	✘ no converter
stack	✔	✔	✔	✔	✔
std	✔	✔	✔	—	—
std_mean	✔	✔	✔	—	—
stft	✔	—	—	—	—
sub	✔	✔	✔	✔	✔
sum	✔	✔	✔	⚠ xfail	✔
sum_to_size	✔	✔	✔	✔	✔
svd	✔	—	—	—	—
svd_lowrank	✔	—	—	—	—
t	✔	✔	✔	✔	✔
t_copy	✘ no converter	✘ no converter	✘ no converter	✘ no converter	✘ no converter
take	✔	✔	✔	✔	✔
take_along_dim	✔	✔	✔	✔	✔
tan	✔	✔	✔	⚠ xfail	⚠ xfail
tanh	✔	✔	✔	⚠ xfail	⚠ xfail
tensor_split	✔	✔	✔	✔	✔
tensordot	✔	✔	✔	✔	✔
tile	✔	✔	✔	✔	✔
to	✔	✔	✔	✔	✔
to_sparse	✘ no converter	✘ no converter	✘ no converter	✘ no converter	✘ no converter
topk	✔	✔	✔	✔	✔
torch._scaled_mm	—	—	—	—	—
torch._scaled_mm_v2	—	—	—	—	—
torch.ops.aten._efficient_attention_forward	—	—	—	—	—
torch.ops.aten._flash_attention_forward	—	—	—	—	—
torch.ops.aten._safe_softmax.default	✔	✔	✔	✔	✔
trace	✘ no converter	—	—	✘ no converter	✘ no converter
transpose	✔	✔	✔	✔	✔
transpose_copy	✔	✔	✔	✔	✔
trapezoid	✘ no converter	✘ no converter	✘ no converter	✘ no converter	✘ no converter
trapz	✘ no converter	✘ no converter	✘ no converter	✘ no converter	✘ no converter
triangular_solve	✘ no converter	—	—	—	—
tril	✔	✔	✔	⚠ xfail	✔
tril_indices	—	—	—	✔	✔
triu	✔	✔	✔	⚠ xfail	✔
triu_indices	—	—	—	✔	✔
true_divide	✔	✔	✔	✔	✔
trunc	✔	✔	✔	⚠ xfail	⚠ xfail
unbind	✔	✔	✔	✔	✔
unbind_copy	✔	✔	✔	✔	✔
unflatten	✔	✔	✔	✔	✔
unfold	✔	✔	✔	✔	✔
unfold_copy	✔	✔	✔	✔	✔
unique	✔	✔	✔	✔	✔
unique_consecutive	✔	✔	✔	✔	✔
unravel_index	—	—	—	✔	✔
unsafe_chunk	✔	✔	✔	✔	✔
unsafe_split	✔	✔	✔	✔	✔
unsqueeze	✔	✔	✔	✔	✔
unsqueeze_copy	✔	✔	✔	✔	✔
var	✘ no converter	✘ no converter	✘ no converter	—	—
var_mean	✘ no converter	✘ no converter	✘ no converter	—	—
vdot	✘ no converter	✘ no converter	✘ no converter	✘ no converter	✘ no converter
view	✔	✔	✔	✔	✔
view_as	✘ no converter	✘ no converter	✘ no converter	✘ no converter	✘ no converter
view_as_complex	✔	✔	—	—	—
view_as_real	—	—	—	—	—
view_copy	✔	✔	✔	✔	✔
vsplit	✔	✔	✔	✔	✔
vstack	✔	✔	✔	✔	✔
where	✔	✔	✔	✔	✔
xlogy	✔	✔	✔	⚠ xfail	⚠ xfail
zero_	✘ no converter	✘ no converter	✘ no converter	✘ no converter	✘ no converter
zeros	✔	✔	✔	✔	✔
zeros_like	✔	✔	✔	✔	✔

[runpythonerror] W0429 17:13:32.166000 19465 torch/_native/cutedsl_utils.py:55] CuTeDSL operators require optional Python packages nvidia-cutlass-dsl and apache-tvm-ffi; missing optional dependency nvidia_cutlass_dsl (importlib.util.find_spec(nvidia_cutlass_dsl) failed)

Float32 comparison across export paths

This section compares, for every op that supports float32, whether it passes, is a known failure (xfail), or has no converter for the three export paths side by side.

Float32 op support: default vs tracing vs new-tracing

Op	Default	Tracing	New-tracing
H	✘ no converter	✘ no converter	✘ no converter
T	✔	✔	✔
__getitem__	✔	✔	✔
__radd__	✔	⚠ xfail	✔
__rdiv__	✔	⚠ xfail	✔
__rmatmul__	✔	✔	✔
__rmod__	✔	✔	✔
__rmul__	✔	⚠ xfail	✔
__rpow__	✔	✔	✔
__rsub__	✘ no converter	✘ no converter	✘ no converter
_batch_norm_with_update	✔	✔	✔
_chunk_cat	✔	✔	✔
_native_batch_norm_legit	✔	✔	✔
_softmax_backward_data	✔	✔	✔
_unsafe_masked_index	✔	✔	✔
_unsafe_masked_index_put_accumulate	✔	✔	✔
_upsample_bilinear2d_aa	✔	✔	✔
abs	✔	✔	✔
acos	✔	✔	✔
acosh	✔	✔	✔
add	✔	✔	✔
addbmm	✔	✔	✔
addcdiv	✘ no converter	✘ no converter	✘ no converter
addcmul	✔	✔	✔
addmm	✔	✔	✔
addmv	✔	✔	✔
addr	✘ no converter	✘ no converter	✘ no converter
alias_copy	✘ no converter	✘ no converter	✘ no converter
all	✔	✔	✔
allclose	✔	✔	✔
amax	✔	✔	✔
amin	✔	✔	✔
aminmax	✔	✔	✔
angle	✔	✔	✔
any	✔	✔	✔
arange	✔	✔	✔
argmax	✔	✔	✔
argmin	✔	✔	✔
argsort	✔	✔	⚠ xfail
argwhere	✔	✔	✔
as_strided	✔	✔	✔
as_strided_copy	✔	✔	✔
as_strided_scatter	✔	✔	✔
asin	✔	✔	✔
asinh	✔	✔	✔
atan	✔	✔	✔
atan2	✔	✔	✔
atanh	✔	✔	✔
atleast_1d	✔	✔	✔
atleast_2d	✔	✔	✔
atleast_3d	✔	✔	✔
baddbmm	✔	✔	✔
bernoulli	✘ no converter	✘ no converter	✘ no converter
bfloat16	⚠ xfail	⚠ xfail	⚠ xfail
block_diag	✔	✔	✔
bmm	✔	✔	✔
bool	✔	✔	⚠ xfail
broadcast_shapes	✔	✔	✔
broadcast_tensors	⚠ xfail	⚠ xfail	⚠ xfail
broadcast_to	✔	✔	✔
bucketize	✔	✔	✔
byte	✔	✔	⚠ xfail
cartesian_prod	✔	✔	✔
cat	✔	✔	✔
cdist	✔	✔	✔
cdouble	✔	✔	✔
ceil	✔	✔	✔
cfloat	✔	✔	✔
chalf	✔	✔	✔
char	✔	✔	⚠ xfail
cholesky	✔	✔	✔
cholesky_inverse	⚠ xfail	⚠ xfail	⚠ xfail
cholesky_solve	⚠ xfail	⚠ xfail	⚠ xfail
chunk	✔	✔	✔
clamp	⚠ xfail	⚠ xfail	⚠ xfail
clamp_max	✔	✔	✔
clamp_min	✔	✔	✔
clone	✔	✔	✔
column_stack	✔	✔	✔
combinations	✔	✔	✔
complex	✔	✔	✔
conj	✔	⚠ xfail	✔
conj_physical	✘ no converter	✘ no converter	✘ no converter
constant_pad_nd	✔	✔	✔
contiguous	✔	✔	✔
copysign	✘ no converter	✘ no converter	✘ no converter
corrcoef	⚠ xfail	⚠ xfail	⚠ xfail
cos	✔	✔	✔
cosh	✔	✔	✔
count_nonzero	✔	✔	✔
cov	⚠ xfail	⚠ xfail	⚠ xfail
cross	✘ no converter	✘ no converter	✘ no converter
cummax	✔	✔	✔
cummin	✔	✔	✔
cumprod	✔	✔	✔
cumsum	✔	✔	✔
cumulative_trapezoid	✘ no converter	✘ no converter	✘ no converter
deg2rad	✘ no converter	✘ no converter	✘ no converter
diag	✔	✔	✔
diag_embed	✔	✔	✔
diagflat	✘ no converter	✘ no converter	✘ no converter
diagonal	✘ no converter	✘ no converter	✘ no converter
diagonal_copy	✘ no converter	✘ no converter	✘ no converter
diagonal_scatter	✘ no converter	✘ no converter	✘ no converter
diff	✔	✔	✔
digamma	✘ no converter	✘ no converter	✘ no converter
dist	✔	✔	✔
dot	✘ no converter	✘ no converter	✘ no converter
double	✔	✔	⚠ xfail
dsplit	✔	✔	✔
dstack	✔	✔	✔
einsum	✔	✔	✔
empty_permuted	✔	✔	✔
empty_strided	✔	✔	✔
eq	✔	✔	✔
equal	✔	✔	✔
erf	✔	✔	✔
erfc	✔	✔	✔
erfinv	✔	✔	⚠ xfail
exp	✔	✔	✔
exp2	✘ no converter	✘ no converter	✘ no converter
expand	✔	✔	✔
expand_as	✔	✔	✔
expand_copy	✔	✔	✔
expm1	✔	✔	✔
eye	✔	✔	✔
fft.fft	✔	✔	✔
fft.fft2	✔	✔	✔
fft.fftn	✔	✔	✔
fft.fftshift	✔	✔	✔
fft.hfft	✔	✔	✔
fft.hfft2	✔	✔	✔
fft.hfftn	✔	✔	✔
fft.ifft	✔	✔	✔
fft.ifft2	✔	✔	✔
fft.ifftn	✔	✔	✔
fft.ifftshift	✔	✔	✔
fft.ihfft	✔	✔	✔
fft.ihfft2	✔	✔	✔
fft.ihfftn	✔	✔	✔
fft.irfft	✔	✔	✔
fft.irfft2	✔	✔	✔
fft.irfftn	✔	✔	✔
fft.rfft	✔	✔	✔
fft.rfft2	✔	✔	✔
fft.rfftn	✔	✔	✔
fill	✔	✔	✔
flatten	✔	⚠ xfail	✔
flip	✔	✔	✔
fliplr	✔	✔	✔
flipud	✔	✔	✔
float	✔	✔	⚠ xfail
float_power	✔	✔	✔
floor	✔	✔	✔
floor_divide	✔	✔	✔
fmax	✔	✔	✔
fmin	✔	✔	✔
fmod	✔	✔	✔
frac	✔	✔	✔
frexp	✔	✔	✔
full	✔	✔	✔
full_like	✔	✔	✔
gather	✔	✔	✔
ge	✔	✔	✔
geqrf	✔	✔	✔
gradient	✔	✔	✔
grid_sampler_2d	✔	✔	✔
grid_sampler_3d	✔	✔	✔
gt	✔	✔	✔
half	✔	✔	⚠ xfail
hash_tensor	✘ no converter	✘ no converter	✘ no converter
heaviside	✔	✔	✔
histc	✔	✔	✔
histogram	✔	✔	✔
histogramdd	✔	✔	✔
hsplit	✔	✔	✔
hstack	✔	✔	✔
hypot	✘ no converter	✘ no converter	✘ no converter
i0	✘ no converter	✘ no converter	✘ no converter
igamma	✘ no converter	✘ no converter	✘ no converter
igammac	✘ no converter	✘ no converter	✘ no converter
index_add	✔	✔	✔
index_copy	✔	✔	✔
index_fill	✔	✔	✔
index_put	✔	✔	✔
index_select	✔	✔	✔
inner	✘ no converter	✘ no converter	✘ no converter
int	✔	✔	⚠ xfail
isclose	✔	✔	⚠ xfail
isfinite	✔	✔	⚠ xfail
isin	✔	✔	✔
isinf	✔	✔	✔
isnan	✔	✔	✔
isneginf	✔	✔	✔
isposinf	✔	✔	✔
isreal	✘ no converter	✘ no converter	✘ no converter
item	✔	✔	✔
jiterator_2inputs_2outputs	⚠ xfail	⚠ xfail	⚠ xfail
jiterator_4inputs_with_extra_args	⚠ xfail	⚠ xfail	⚠ xfail
jiterator_binary	⚠ xfail	⚠ xfail	⚠ xfail
jiterator_binary_return_by_ref	⚠ xfail	⚠ xfail	⚠ xfail
jiterator_unary	⚠ xfail	⚠ xfail	⚠ xfail
kron	✘ no converter	✘ no converter	✘ no converter
kthvalue	✔	✔	✔
ldexp	✘ no converter	✘ no converter	✘ no converter
le	✔	✔	✔
lerp	✔	✔	✔
lgamma	✘ no converter	✘ no converter	✘ no converter
linalg.cholesky	✔	✔	✔
linalg.cholesky_ex	✔	✔	✔
linalg.cond	✘ no converter	✘ no converter	✘ no converter
linalg.cross	✔	✔	✔
linalg.det	✔	✔	✔
linalg.diagonal	✘ no converter	✘ no converter	✘ no converter
linalg.eig	✔	✔	✔
linalg.eigh	✔	✔	✔
linalg.eigvals	✔	✔	✔
linalg.eigvalsh	✔	✔	✔
linalg.householder_product	✘ no converter	✘ no converter	✘ no converter
linalg.inv	✘ no converter	✘ no converter	✘ no converter
linalg.inv_ex	✘ no converter	✘ no converter	✘ no converter
linalg.ldl_factor	✔	✔	✔
linalg.ldl_factor_ex	✔	✔	✔
linalg.ldl_solve	✔	✔	✔
linalg.lstsq	✔	✔	✔
linalg.lu	✔	✔	✔
linalg.lu_factor	✔	✔	✔
linalg.lu_factor_ex	✔	✔	✔
linalg.lu_solve	✔	✔	✔
linalg.matrix_norm	✔	✔	✔
linalg.matrix_power	✔	✔	✔
linalg.matrix_rank	✔	✔	✔
linalg.multi_dot	✔	✔	✔
linalg.norm	✔	✔	✔
linalg.pinv	✔	✔	✔
linalg.qr	✘ no converter	✘ no converter	✘ no converter
linalg.slogdet	✔	✔	✔
linalg.solve	✘ no converter	✘ no converter	✘ no converter
linalg.solve_ex	✘ no converter	✘ no converter	✘ no converter
linalg.solve_triangular	✔	✔	✔
linalg.svd	✔	✔	✔
linalg.svdvals	✘ no converter	✘ no converter	✘ no converter
linalg.tensorinv	✔	✔	✔
linalg.tensorsolve	✔	✔	✔
linalg.vander	✔	✔	✔
linalg.vecdot	✔	✔	✔
linalg.vector_norm	✔	✔	✔
linspace	✔	✔	✔
log	✔	✔	✔
log10	✔	✔	✔
log1p	✔	✔	✔
log2	✔	✔	✔
log_softmax	✔	✔	✔
logaddexp	✔	✔	✔
logaddexp2	✔	✔	✔
logcumsumexp	✔	✔	✔
logdet	✔	✔	✔
logical_and	✔	⚠ xfail	✔
logical_not	⚠ xfail	⚠ xfail	⚠ xfail
logical_or	✔	⚠ xfail	✔
logical_xor	✔	⚠ xfail	✔
logit	✔	✔	✔
logspace	✔	✔	✔
logsumexp	✔	✔	✔
long	✔	✔	⚠ xfail
lt	✔	✔	✔
lu	✔	✔	✔
lu_solve	✘ no converter	✘ no converter	✘ no converter
lu_unpack	✘ no converter	✘ no converter	✘ no converter
mH	✔	✔	✔
mT	✔	✔	✔
masked.amax	✔	✔	✔
masked.amin	✔	✔	✔
masked.argmax	✔	✔	✔
masked.argmin	✔	✔	✔
masked.cumprod	✔	✔	✔
masked.cumsum	✔	✔	✔
masked.log_softmax	✔	✔	✔
masked.logaddexp	✔	✔	✔
masked.logsumexp	✔	✔	✔
masked.mean	✔	✔	✔
masked.median	✔	✔	✔
masked.norm	✔	✔	✔
masked.normalize	✔	✔	✔
masked.prod	✔	✔	✔
masked.softmax	✔	✔	✔
masked.softmin	✔	✔	✔
masked.std	✔	✔	✔
masked.sum	✔	✔	✔
masked.var	✔	✔	✔
masked_fill	✔	✔	✔
masked_scatter	✔	✔	✔
masked_select	✘ no converter	✘ no converter	✘ no converter
matmul	✔	✔	✔
matrix_exp	✘ no converter	✘ no converter	✘ no converter
max_pool2d_with_indices_backward	✔	✔	✔
maximum	✔	✔	✔
mean	✔	✔	✔
median	✘ no converter	✘ no converter	✘ no converter
minimum	✔	✔	✔
mm	✔	✔	✔
mode	✘ no converter	✘ no converter	✘ no converter
movedim	✔	✔	✔
msort	✘ no converter	✘ no converter	✘ no converter
mul	✔	✔	✔
multinomial	✔	✔	✔
mv	✘ no converter	✘ no converter	✘ no converter
nan_to_num	⚠ xfail	⚠ xfail	⚠ xfail
nanmean	✔	✔	⚠ xfail
nanmedian	✘ no converter	✘ no converter	✘ no converter
nanquantile	✔	✔	✔
nansum	✔	✔	✔
narrow	✔	✔	✔
narrow_copy	✔	✔	✔
native_batch_norm	✔	✔	✔
native_dropout_backward	✔	✔	✔
native_layer_norm	✔	✔	✔
ne	✔	✔	✔
neg	✔	✔	✔
new_empty	✔	✔	✔
new_empty_strided	✔	✔	✔
new_full	✔	✔	✔
new_ones	✔	✔	✔
new_zeros	✔	✔	✔
nextafter	✘ no converter	✘ no converter	✘ no converter
nn.functional.adaptive_avg_pool1d	✔	✔	✔
nn.functional.adaptive_avg_pool2d	✔	✔	✔
nn.functional.adaptive_avg_pool3d	✔	✔	✔
nn.functional.adaptive_max_pool1d	✔	✔	✔
nn.functional.adaptive_max_pool2d	✔	✔	✔
nn.functional.adaptive_max_pool3d	✔	✔	✔
nn.functional.alpha_dropout	✔	✔	✔
nn.functional.avg_pool1d	✔	✔	✔
nn.functional.avg_pool2d	✔	✔	✔
nn.functional.avg_pool3d	✔	✔	✔
nn.functional.batch_norm	✔	✔	✔
nn.functional.bilinear	✔	⚠ xfail	⚠ xfail
nn.functional.binary_cross_entropy	✔	✔	✔
nn.functional.binary_cross_entropy_with_logits	✔	✔	✔
nn.functional.celu	✔	✔	✔
nn.functional.channel_shuffle	✔	✔	✔
nn.functional.conv1d	✔	✔	✔
nn.functional.conv2d	✔	✔	✔
nn.functional.conv3d	✔	✔	✔
nn.functional.conv_transpose1d	✔	✔	✔
nn.functional.conv_transpose2d	✔	✔	✔
nn.functional.conv_transpose3d	✔	✔	✔
nn.functional.cosine_embedding_loss	✔	✔	✔
nn.functional.cosine_similarity	✔	✔	✔
nn.functional.cross_entropy	✔	✔	✔
nn.functional.ctc_loss	✔	✔	✔
nn.functional.dropout	✔	✔	✔
nn.functional.dropout2d	✔	✔	✔
nn.functional.dropout3d	✔	✔	✔
nn.functional.elu	✔	✔	✔
nn.functional.embedding	✔	✔	✔
nn.functional.embedding_bag	✔	✔	✔
nn.functional.fractional_max_pool2d	✔	✔	✔
nn.functional.fractional_max_pool3d	✔	✔	✔
nn.functional.gaussian_nll_loss	✔	✔	✔
nn.functional.gelu	✔	✔	✔
nn.functional.glu	✔	✔	✔
nn.functional.grid_sample	✔	✔	✔
nn.functional.group_norm	✔	✔	✔
nn.functional.hardshrink	✔	✔	✔
nn.functional.hardsigmoid	✔	⚠ xfail	✔
nn.functional.hardswish	✔	✔	✔
nn.functional.hardtanh	✔	✔	✔
nn.functional.hinge_embedding_loss	✔	✔	✔
nn.functional.huber_loss	✔	✔	✔
nn.functional.instance_norm	✔	✔	✔
nn.functional.kl_div	✔	✔	✔
nn.functional.l1_loss	✔	✔	✔
nn.functional.layer_norm	✔	✔	✔
nn.functional.leaky_relu	✔	✔	✔
nn.functional.linear	✔	✔	✔
nn.functional.local_response_norm	✔	✔	✔
nn.functional.logsigmoid	✔	✔	✔
nn.functional.margin_ranking_loss	✔	✔	✔
nn.functional.max_pool1d	✔	✔	✔
nn.functional.max_pool2d	✔	✔	✔
nn.functional.max_pool3d	✔	✔	✔
nn.functional.max_unpool1d	✔	✔	✔
nn.functional.max_unpool2d	✔	✔	✔
nn.functional.max_unpool3d	✔	✔	✔
nn.functional.mish	✔	✔	✔
nn.functional.mse_loss	✔	✔	✔
nn.functional.multi_head_attention_forward	✔	✔	✔
nn.functional.multi_margin_loss	✘ no converter	✘ no converter	✘ no converter
nn.functional.multilabel_margin_loss	✘ no converter	✘ no converter	✘ no converter
nn.functional.multilabel_soft_margin_loss	✘ no converter	✘ no converter	✘ no converter
nn.functional.nll_loss	✔	✔	✔
nn.functional.normalize	✔	✔	✔
nn.functional.pairwise_distance	✔	✔	✔
nn.functional.pdist	✘ no converter	✘ no converter	✘ no converter
nn.functional.pixel_shuffle	✔	✔	✔
nn.functional.pixel_unshuffle	✔	✔	✔
nn.functional.poisson_nll_loss	✔	✔	✔
nn.functional.prelu	✔	✔	✔
nn.functional.relu	✔	✔	✔
nn.functional.relu6	✔	✔	✔
nn.functional.rms_norm	✔	✔	✔
nn.functional.rrelu	✔	✔	✔
nn.functional.scaled_dot_product_attention	✔	✔	✔
nn.functional.selu	✔	✔	✔
nn.functional.silu	✔	✔	✔
nn.functional.smooth_l1_loss	✔	✔	✔
nn.functional.soft_margin_loss	✔	✔	✔
nn.functional.softmin	✔	✔	✔
nn.functional.softplus	✔	✔	✔
nn.functional.softshrink	✔	✔	✔
nn.functional.softsign	✔	✔	✔
nn.functional.tanhshrink	✔	✔	✔
nn.functional.threshold	✔	✔	✔
nn.functional.triplet_margin_loss	✔	✔	✔
nn.functional.triplet_margin_with_distance_loss	✔	✔	✔
nn.functional.unfold	✔	✔	✔
nn.functional.upsample_bilinear	✔	✔	✔
nn.functional.upsample_nearest	✔	✔	✔
nonzero	✔	✔	✔
nonzero_static	✔	✔	✔
norm	✔	✔	✔
ones	✔	✔	✔
ones_like	✔	✔	✔
ormqr	✔	✔	✔
outer	✔	✔	✔
pca_lowrank	✔	✔	✔
permute	✔	✔	✔
permute_copy	✔	✔	✔
pinverse	✘ no converter	✘ no converter	✘ no converter
polar	✔	✔	✔
positive	✘ no converter	✘ no converter	✘ no converter
pow	✔	✔	✔
prod	✔	✔	✔
put	✔	✔	✔
qr	✘ no converter	✘ no converter	✘ no converter
quantile	✔	✔	✔
rad2deg	✘ no converter	✘ no converter	✘ no converter
randint_like	✔	✔	✔
ravel	✔	✔	✔
real	✘ no converter	✘ no converter	✘ no converter
reciprocal	✔	✔	✔
remainder	✔	✔	✔
renorm	✔	✔	✔
repeat	✔	✔	✔
repeat_interleave	✔	✔	✔
reshape	✔	✔	✔
reshape_as	✔	✔	✔
resize_	✔	✔	✔
resize_as_	✘ no converter	✘ no converter	✘ no converter
resolve_conj	✘ no converter	✘ no converter	✘ no converter
resolve_neg	✘ no converter	✘ no converter	✘ no converter
roll	✔	✔	✔
rot90	✔	✔	✔
round	✔	✔	✔
rsqrt	✔	✔	✔
rsub	✘ no converter	✘ no converter	✘ no converter
scalar_tensor	✔	✔	✔
scatter	✔	✔	✔
scatter_add	✔	✔	✔
searchsorted	✔	✔	✔
select	✔	✔	✔
select_scatter	✔	✔	✔
sgn	✘ no converter	✘ no converter	✘ no converter
short	✔	✔	⚠ xfail
sigmoid	✔	✔	✔
sign	✔	✔	✔
signal.windows.bartlett	✔	✔	✔
signal.windows.blackman	✔	✔	✔
signal.windows.cosine	✔	✔	✔
signal.windows.exponential	✔	✔	✔
signal.windows.gaussian	✔	✔	✔
signal.windows.general_cosine	✔	✔	✔
signal.windows.general_hamming	✔	✔	✔
signal.windows.hamming	✔	✔	✔
signal.windows.hann	✔	✔	✔
signal.windows.kaiser	✔	✔	✔
signal.windows.nuttall	✔	✔	✔
signbit	✔	✔	✔
sin	✔	✔	✔
sinc	✘ no converter	✘ no converter	✘ no converter
sinh	✔	✔	✔
slice	✔	✔	✔
slice_scatter	✔	✔	✔
softmax	✔	✔	✔
sort	✘ no converter	✘ no converter	✘ no converter
sparse.sampled_addmm	✔	✔	✔
special.airy_ai	✘ no converter	✘ no converter	✘ no converter
special.bessel_j0	✘ no converter	✘ no converter	✘ no converter
special.bessel_j1	✘ no converter	✘ no converter	✘ no converter
special.bessel_y0	✘ no converter	✘ no converter	✘ no converter
special.bessel_y1	✘ no converter	✘ no converter	✘ no converter
special.chebyshev_polynomial_t	✘ no converter	✘ no converter	✘ no converter
special.chebyshev_polynomial_u	✘ no converter	✘ no converter	✘ no converter
special.chebyshev_polynomial_v	✘ no converter	✘ no converter	✘ no converter
special.chebyshev_polynomial_w	✘ no converter	✘ no converter	✘ no converter
special.entr	✘ no converter	✘ no converter	✘ no converter
special.erfcx	✘ no converter	✘ no converter	✘ no converter
special.hermite_polynomial_h	✘ no converter	✘ no converter	✘ no converter
special.hermite_polynomial_he	✘ no converter	✘ no converter	✘ no converter
special.i0e	✘ no converter	✘ no converter	✘ no converter
special.i1	✘ no converter	✘ no converter	✘ no converter
special.i1e	✘ no converter	✘ no converter	✘ no converter
special.laguerre_polynomial_l	✘ no converter	✘ no converter	✘ no converter
special.legendre_polynomial_p	✘ no converter	✘ no converter	✘ no converter
special.log_ndtr	✘ no converter	✘ no converter	✘ no converter
special.modified_bessel_i0	✘ no converter	✘ no converter	✘ no converter
special.modified_bessel_i1	✘ no converter	✘ no converter	✘ no converter
special.modified_bessel_k0	✘ no converter	✘ no converter	✘ no converter
special.modified_bessel_k1	✘ no converter	✘ no converter	✘ no converter
special.ndtr	✘ no converter	✘ no converter	✘ no converter
special.ndtri	✘ no converter	✘ no converter	✘ no converter
special.scaled_modified_bessel_k0	✘ no converter	✘ no converter	✘ no converter
special.scaled_modified_bessel_k1	✘ no converter	✘ no converter	✘ no converter
special.shifted_chebyshev_polynomial_t	✘ no converter	✘ no converter	✘ no converter
special.shifted_chebyshev_polynomial_u	✘ no converter	✘ no converter	✘ no converter
special.shifted_chebyshev_polynomial_v	✘ no converter	✘ no converter	✘ no converter
special.shifted_chebyshev_polynomial_w	✘ no converter	✘ no converter	✘ no converter
special.spherical_bessel_j0	✘ no converter	✘ no converter	✘ no converter
special.xlog1py	✘ no converter	✘ no converter	✘ no converter
special.zeta	✘ no converter	✘ no converter	✘ no converter
split	✔	✔	✔
split_with_sizes	✔	✔	✔
split_with_sizes_copy	✔	✔	✔
sqrt	✔	✔	✔
square	✔	✔	✔
squeeze	✔	✔	✔
squeeze_copy	✘ no converter	✘ no converter	✘ no converter
stack	✔	✔	✔
std	✔	⚠ xfail	⚠ xfail
std_mean	✔	⚠ xfail	⚠ xfail
stft	✔	✔	✔
sub	✔	✔	✔
sum	✔	✔	✔
sum_to_size	✔	✔	✔
svd	✔	✔	✔
svd_lowrank	✔	✔	✔
t	✔	✔	✔
t_copy	✘ no converter	✘ no converter	✘ no converter
take	✔	✔	✔
take_along_dim	✔	✔	✔
tan	✔	✔	✔
tanh	✔	✔	✔
tensor_split	✔	✔	⚠ xfail
tensordot	✔	✔	✔
tile	✔	✔	✔
to	✔	✔	✔
to_sparse	✘ no converter	✘ no converter	✘ no converter
topk	✔	✔	✔
torch.ops.aten._safe_softmax.default	✔	✔	✔
trace	✘ no converter	✘ no converter	✘ no converter
transpose	✔	✔	✔
transpose_copy	✔	✔	✔
trapezoid	✘ no converter	✘ no converter	✘ no converter
trapz	✘ no converter	✘ no converter	✘ no converter
triangular_solve	✘ no converter	✘ no converter	✘ no converter
tril	✔	✔	✔
triu	✔	✔	✔
true_divide	✔	✔	✔
trunc	✔	✔	✔
unbind	✔	✔	✔
unbind_copy	✔	✔	✔
unflatten	✔	✔	✔
unfold	✔	✔	✔
unfold_copy	✔	✔	✔
unique	✔	✔	✔
unique_consecutive	✔	✔	✔
unsafe_chunk	✔	✔	✔
unsafe_split	✔	✔	✔
unsqueeze	✔	✔	✔
unsqueeze_copy	✔	✔	✔
var	✘ no converter	✘ no converter	✘ no converter
var_mean	✘ no converter	✘ no converter	✘ no converter
vdot	✘ no converter	✘ no converter	✘ no converter
view	✔	✔	✔
view_as	✘ no converter	✘ no converter	✘ no converter
view_as_complex	✔	✔	✔
view_copy	✔	✔	✔
vsplit	✔	✔	✔
vstack	✔	✔	✔
where	✔	✔	✔
xlogy	✔	✔	✔
zero_	✘ no converter	✘ no converter	✘ no converter
zeros	✔	✔	✔
zeros_like	✔	✔	✔

[runpythonerror] W0429 17:13:48.096000 19811 torch/_native/cutedsl_utils.py:55] CuTeDSL operators require optional Python packages nvidia-cutlass-dsl and apache-tvm-ffi; missing optional dependency nvidia_cutlass_dsl (importlib.util.find_spec(nvidia_cutlass_dsl) failed)