torch-mlir

Author	SHA1	Message	Date
Jiahao Li	eaab9be207	Add e2e support for aten logical or/and/xor/not ops (#1752 )	2022-12-26 10:23:38 +08:00
Jiahao Li	60a139271d	Add aten.std.correction op and its decomposition (#1731 )	2022-12-21 21:02:40 -08:00
Jiahao Li	15b249777b	[Torch][MHLO] Decompose aten.copy op. Lower aten.rsqrt & sigmoid to mhlo. (#1734 )	2022-12-22 10:13:59 +08:00
ataheridezfouli-groq	17ee643aeb	[TORCH] Add Complex Number support (#1673 ) Add Complex number dtype support to torch tensors. Add aten.fft_fft op to test complex numbers.	2022-12-15 21:40:01 +00:00
Ramiro Leal-Cavazos	a710237437	[custom op] Generalize shape library logic to work with dtypes (#1594 ) * [custom op] Generalize shape library logic to work with dtypes This commit generalizes the shape library logic, so that dtype rules for ops can also be expressed using the same mechanism. In other words, each op can now have a shape function and a dtype function specified in Python that is imported during lowering to calculate the shapes and dtypes throught a program. For more information about how to specify a dtype function, see the updated `docs/adding_a_shape_and_dtype_function.md`. For those not familiar with how the shape library works, the file `docs/calculations_lib.md` provides an overview.	2022-12-13 08:25:41 -08:00