Jun Luo 00e19ae97a [MTIA] Support module.mtia() (#131499) ... Summary: Following other device backends' implementation to support module.mtia() API. Test Plan: OSS and Internal CIs. Differential Revision: D60076584 Pull Request resolved: https://github.com/pytorch/pytorch/pull/131499 Approved by: https://github.com/mikaylagawarecki		2024-07-25 04:23:48 +00:00
..
attention	Add sparse block to flex_decoding kernel (#130884)	2024-07-24 20:30:25 +00:00
backends
intrinsic	[BE][Easy] enable UFMT for torch/nn/ (#128865)	2024-07-25 02:48:42 +00:00
modules	[MTIA] Support module.mtia() (#131499)	2024-07-25 04:23:48 +00:00
parallel	[BE][Easy] enable UFMT for torch/nn/ (#128865)	2024-07-25 02:48:42 +00:00
qat	[BE][Easy] enable UFMT for torch/nn/ (#128865)	2024-07-25 02:48:42 +00:00
quantizable	[BE][Easy] enable UFMT for torch/nn/ (#128865)	2024-07-25 02:48:42 +00:00
quantized	[BE][Easy] enable UFMT for torch/nn/ (#128865)	2024-07-25 02:48:42 +00:00
utils	[BE] mypy: disallow untyped decorators (#131428)	2024-07-23 21:50:55 +00:00
__init__.py
_reduction.py
common_types.py
cpp.py
functional.py	Fix an example: Resolve broadcasting error in attn_bias and attn_mask… (#130209)	2024-07-19 15:23:22 +00:00
functional.pyi.in	[torchgen] reference generated comment to actual location of the generator and template (#130020)	2024-07-05 21:47:14 +00:00
grad.py
init.py	docs: note transposed weight initialisations (#130122)	2024-07-19 15:23:03 +00:00
parameter.py	Only test _is_param if doing instance check on Parameter base (#130578)	2024-07-12 13:55:13 +00:00
parameter.pyi	Use Generic TypeAlias (PEP 585) and Union Type (PEP 604) in .pyi stub files (#129419)	2024-06-29 09:23:39 +00:00