c036fb3e7d
assert lambda >= 0 in poisson distribution cuda kernel ( #85906 )
...
fix https://github.com/pytorch/pytorch/issues/85731
Pull Request resolved: https://github.com/pytorch/pytorch/pull/85906
Approved by: https://github.com/ngimel
2022-09-29 22:11:00 +00:00
a0b1693996
Revert "Update amax/amin/norm/count_nonzero signatures with int[*]? dim ( #83300 )"
...
This reverts commit 1c0f0b33a0e013d6ec162cf488ff7643c4ffa33e.
Reverted https://github.com/pytorch/pytorch/pull/83300 on behalf of https://github.com/jeffdaily due to The commit breaks nvfuser tests
2022-09-28 17:04:53 +00:00
1c0f0b33a0
Update amax/amin/norm/count_nonzero signatures with int[*]? dim ( #83300 )
...
Changes `dim` arg to use `int[*]?` type for the following functions in `native_funcitons.yaml`:
* `amax`
* `amin`
* `norm`
* `frobenius_norm`
* `native_norm`
* `count_nonzero`
Part of #29137
Pull Request resolved: https://github.com/pytorch/pytorch/pull/83300
Approved by: https://github.com/ngimel , https://github.com/albanD , https://github.com/kulinseth
2022-09-28 01:56:37 +00:00
572dd862c4
Revert "Update amax/amin/norm/count_nonzero signatures with int[*]? dim ( #83300 )"
...
This reverts commit 8c7c7ed3221aeeefb63ef2b7a221a5d8b274cda5.
Reverted https://github.com/pytorch/pytorch/pull/83300 on behalf of https://github.com/huydhn due to The commit pin breaks XLA test somehow
2022-09-28 01:36:43 +00:00
9858f04350
[quant][docs] Add types for scale and zero_point tensor for torch.fake_quantize_per_channel_affine docs ( #85733 )
...
Summary:
Fixes: https://github.com/pytorch/pytorch/issues/85525
Test Plan:
visual inspection for the docs page
Reviewers:
Subscribers:
Tasks:
Tags:
Pull Request resolved: https://github.com/pytorch/pytorch/pull/85733
Approved by: https://github.com/z-a-f
2022-09-28 00:43:48 +00:00
8c7c7ed322
Update amax/amin/norm/count_nonzero signatures with int[*]? dim ( #83300 )
...
Changes `dim` arg to use `int[*]?` type for the following functions in `native_funcitons.yaml`:
* `amax`
* `amin`
* `norm`
* `frobenius_norm`
* `native_norm`
* `count_nonzero`
Part of #29137
Pull Request resolved: https://github.com/pytorch/pytorch/pull/83300
Approved by: https://github.com/ngimel , https://github.com/albanD , https://github.com/kulinseth
2022-09-27 23:50:04 +00:00
e1f9125e61
[doc] add argument default values in rot90 ( #85610 )
...
Add argument default values in rot90.
Pull Request resolved: https://github.com/pytorch/pytorch/pull/85610
Approved by: https://github.com/lezcano
2022-09-25 17:10:45 +00:00
bd5efbb7ee
[doc] add pin_memory argument to rand ( #85221 )
...
Similar to #85123
cc - @mruberry @kshitij12345
Pull Request resolved: https://github.com/pytorch/pytorch/pull/85221
Approved by: https://github.com/mruberry
2022-09-25 10:47:59 +00:00
539076e2c2
Remove deprecated torch.lstsq ( #70980 )
...
The time has come to remove deprecated linear algebra related functions. This PR removes `torch.lstsq`.
There's a note in `tools/codegen/gen.py` about `lstsq` schema in `native_function.yaml` that I will not remove:
87139d8532/tools/codegen/gen.py (L734-L770)https://github.com/pytorch/pytorch/pull/70980
Approved by: https://github.com/lezcano , https://github.com/kit1980
2022-09-23 00:16:55 +00:00
bcf93181a0
Remove deprecated torch.matrix_rank ( #70981 )
...
The time has come to remove deprecated linear algebra related functions. This PR removes `torch.matrix_rank`.
cc @jianyuh @nikitaved @pearu @mruberry @walterddr @IvanYashchuk @xwang233 @Lezcano
Pull Request resolved: https://github.com/pytorch/pytorch/pull/70981
Approved by: https://github.com/lezcano , https://github.com/kit1980
2022-09-22 17:40:46 +00:00
6fb182c86b
[doc] document pin_memory for randn ( #85219 )
...
Fixes #85123
cc: @mruberry @kshitij12345
Pull Request resolved: https://github.com/pytorch/pytorch/pull/85219
Approved by: https://github.com/mruberry
2022-09-21 22:43:52 +00:00
3a09a1e8f0
[doc] remove out argument from squeeze ( #85222 )
...
Fixes #83972
cc- @ngimel @kshitij12345
Pull Request resolved: https://github.com/pytorch/pytorch/pull/85222
Approved by: https://github.com/ezyang
2022-09-21 19:31:56 +00:00
2a88f1b2d8
Land "Make ceil,floor,round,trunc handle integers" ( #85144 )
...
PR to land https://github.com/pytorch/pytorch/pull/78480 , as Rohit does
not work in the PyTorch project anymore
Pull Request resolved: https://github.com/pytorch/pytorch/pull/85144
Approved by: https://github.com/ngimel , https://github.com/mruberry
2022-09-21 17:23:47 +00:00
7234eb06f7
Revert "Land "Make ceil,floor,round,trunc handle integers" ( #85144 )"
...
This reverts commit b27eb8d377fc8ac267fdaed7f95a03d609764604.
Reverted https://github.com/pytorch/pytorch/pull/85144 on behalf of https://github.com/clee2000 due to broke slow tests in trunk ex https://ossci-raw-job-status.s3.amazonaws.com/log/8433956087
2022-09-19 18:46:35 +00:00
b27eb8d377
Land "Make ceil,floor,round,trunc handle integers" ( #85144 )
...
PR to land https://github.com/pytorch/pytorch/pull/78480 , as Rohit does
not work in the PyTorch project anymore
Pull Request resolved: https://github.com/pytorch/pytorch/pull/85144
Approved by: https://github.com/ngimel , https://github.com/mruberry
2022-09-19 17:21:48 +00:00
3a51b557ef
Added docs and opinfo for narrow_copy ( #84493 )
...
Pull Request resolved: https://github.com/pytorch/pytorch/pull/84493
Approved by: https://github.com/amjames , https://github.com/ngimel , https://github.com/mruberry
2022-09-19 14:28:25 +00:00
735154354b
update torch.narrow doc ( #85180 )
...
Fixes #84783
Pull Request resolved: https://github.com/pytorch/pytorch/pull/85180
Approved by: https://github.com/kit1980
2022-09-16 21:44:23 +00:00
2386cd2945
[reland] [numpy] add torch.concatenate, alias of torch.cat ( #85073 )
...
Previous PR: #82946
Fixes #81161
Pull Request resolved: https://github.com/pytorch/pytorch/pull/85073
Approved by: https://github.com/mruberry
2022-09-15 19:34:44 +00:00
fa7bf3e2dc
Revert "[numpy] add torch.concatenate, alias of torch.cat ( #82946 )"
...
This reverts commit 270e5e519d98868af0166f3a179b286682cfb267.
Reverted https://github.com/pytorch/pytorch/pull/82946 on behalf of https://github.com/malfet due to Broke M1 tests, see 270e5e519d
2022-09-14 21:32:11 +00:00
270e5e519d
[numpy] add torch.concatenate, alias of torch.cat ( #82946 )
...
As per the title. Fixes : #81161
- [x] add ErrorInputs
- ~[ ] dtype argument?~
- ~[ ] casting argument?~
As discussed offline with @kshitij12345, we can currently ignore `dtype` and `casting` arguments.
cc: @kshitij12345!
Pull Request resolved: https://github.com/pytorch/pytorch/pull/82946
Approved by: https://github.com/mruberry
2022-09-14 19:28:43 +00:00
01c54ad6de
Remove deprecated torch.eig ( #70982 )
...
The time has come to remove deprecated linear algebra related functions. This PR removes `torch.eig`.
cc @jianyuh @nikitaved @pearu @mruberry @walterddr @IvanYashchuk @xwang233 @Lezcano
Pull Request resolved: https://github.com/pytorch/pytorch/pull/70982
Approved by: https://github.com/Lezcano , https://github.com/malfet
2022-09-09 21:31:57 +00:00
9b115c7bd3
Sparse Compressed Transpose add support for Batch dims and BSR/BSC layouts ( #82122 )
...
Pull Request resolved: https://github.com/pytorch/pytorch/pull/82122
Approved by: https://github.com/bhosmer
2022-09-02 17:42:58 +00:00
b136f3f310
More doctest refinements. ( #83317 )
...
Follow up to #82797
Now that the doctests themselves are in a better state, we should be able to enable xdoctest on the CI so they stay that way.
@ezyang @vadimkantorov
Pull Request resolved: https://github.com/pytorch/pytorch/pull/83317
Approved by: https://github.com/ezyang
2022-08-22 20:07:26 +00:00
f534b2c627
Revert "Remove split functional wrapper ( #74727 )"
...
This reverts commit a58876ace78df1cfeb136cad592487f34d7e02f1.
Reverted https://github.com/pytorch/pytorch/pull/74727 on behalf of https://github.com/seemethere due to Fails internal use cases, might extend out to external use cases as well. Need to assess overall impact of this change more widely
2022-08-10 19:45:23 +00:00
a58876ace7
Remove split functional wrapper ( #74727 )
...
Pull Request resolved: https://github.com/pytorch/pytorch/pull/74727
Approved by: https://github.com/albanD , https://github.com/khabinov
2022-08-10 17:57:48 +00:00
be5b3df6cc
Update std_mean/var_mean/nanmean/nansum signatures with int[1]? dim ( #82912 )
...
### Description
Change the type of the `dim` arg for `std_mean/var_mean/nanmean/nansum` to `int[1]?` in `native_functions.yaml`
### Issue
Part of #29137
### Testing
Pull Request resolved: https://github.com/pytorch/pytorch/pull/82912
Approved by: https://github.com/albanD
2022-08-10 16:58:26 +00:00
5ca9b2b6fa
Enable dim=None for torch.var ( #82765 )
...
### Description
Add support for `dim=None` in `torch.var`
### Issue
Part of #29137
### Testing
N/A
Pull Request resolved: https://github.com/pytorch/pytorch/pull/82765
Approved by: https://github.com/albanD
2022-08-04 20:47:27 +00:00
ff5399e528
Revise sparse docs regarding Sparse Compressed tensors ( #82108 )
...
Pull Request resolved: https://github.com/pytorch/pytorch/pull/82108
Approved by: https://github.com/bhosmer
2022-07-29 18:15:09 +00:00
8def154e00
Fix multiple docstring type mistakes ( #82474 )
...
### Description
* Docstrings using `(tuple of ints)` shows up as `(tuple of python:ints)`, so I fixed them by making the `int` no longer plural. Example: https://pytorch.org/docs/stable/generated/torch.permute.html#torch.permute
* A docstring type in JIT had one of its types incorrectly highlighted as code. Example: https://pytorch.org/docs/stable/generated/torch.jit.script.html#torch.jit.script
* I found some docstring type usages of `string` that had not yet been converted to `str` after #82410
* Some docstrings incorrectly listed their defaults inside the docstring types.
* I also found a docstring that was missing its type
### Testing
No testing should be required.
---
In the developer guidelines, there should probably be standards listed for the docstring types.
Pull Request resolved: https://github.com/pytorch/pytorch/pull/82474
Approved by: https://github.com/albanD
2022-07-29 17:45:37 +00:00
fd84c458f4
Add torch.unflatten and improve its docs ( #81399 )
...
unflatten now has a free function version in torch.flatten in addition to
the method in torch.Tensor.flatten.
Updated docs to reflect this and polished them a little.
For consistency, changed the signature of the int version of unflatten in
native_functions.yaml.
Some override tests were failing because unflatten has unusual
characteristics in terms of the .int and .Dimname versions having
different number of arguments so this required some changes
to test/test_override.py
Removed support for using mix of integer and string arguments
when specifying dimensions in unflatten.
Pull Request resolved: https://github.com/pytorch/pytorch/pull/81399
Approved by: https://github.com/Lezcano , https://github.com/ngimel
2022-07-29 15:02:42 +00:00
2bfae07a79
Enable dim=None for torch.mean ( #81286 )
...
Part of #79525
This will require coordination with XLA before merging, just like #79881
Pull Request resolved: https://github.com/pytorch/pytorch/pull/81286
Approved by: https://github.com/albanD
2022-07-28 22:34:56 +00:00
357b7d589c
Fix docstring inconsistencies: string -> str, boolean -> bool ( #82410 )
...
### Description
Throughout the PyTorch docs and codebase, the `string` type in docstrings is referred to by two separate names. This leads to inconsistent docs, like you can see here: https://pytorch.org/docs/stable/generated/torch.nn.Conv3d.html#torch.nn.Conv3d
This PR fixes this issue by ensuring that all mentions of the string type in docstrings, are using the same format that Sphinx generates hyperlinks for.
### Testing
No testing should be required for this change
Pull Request resolved: https://github.com/pytorch/pytorch/pull/82410
Approved by: https://github.com/jbschlosser
2022-07-28 21:29:57 +00:00
026ef78b22
[DOC] fix a mistake in torch.dist document ( #82104 )
...
In document (https://pytorch.org/docs/stable/generated/torch.dist.html )the torch.dist(x, y, 0) returns inf.
However, it should return 4.0 and the pytorch (1.12) does return 4.0
Pull Request resolved: https://github.com/pytorch/pytorch/pull/82104
Approved by: https://github.com/samdow
2022-07-26 13:51:31 +00:00
12cb26509a
Apply ufmt to torch internal ( #81643 )
...
This is a big bang PR, merge conflicts are probably expected and will be addressed at merge.
Pull Request resolved: https://github.com/pytorch/pytorch/pull/81643
Approved by: https://github.com/ezyang
2022-07-22 02:19:50 +00:00
e505796a2c
[Array API] Add linalg.vecdot ( #70542 )
...
This PR adds the function `linalg.vecdot` specified by the [Array
API](https://data-apis.org/array-api/latest/API_specification/linear_algebra_functions.html#function-vecdot )
For the complex case, it chooses to implement \sum x_i y_i. See the
discussion in https://github.com/data-apis/array-api/issues/356
Edit. When it comes to testing, this function is not quite a binopt, nor a reduction opt. As such, we're this close to be able to get the extra testing, but we don't quite make it. Now, it's such a simple op that I think we'll make it without this.
Resolves https://github.com/pytorch/pytorch/issues/18027 .
cc @mruberry @rgommers @pmeier @asmeurer @leofang @AnirudhDagar @asi1024 @emcastillo @kmaehashi
Pull Request resolved: https://github.com/pytorch/pytorch/pull/70542
Approved by: https://github.com/IvanYashchuk , https://github.com/mruberry
2022-07-12 14:28:54 +00:00
7f3677d723
Revert "Remove split functional wrapper ( #74727 )"
...
This reverts commit cc3126083ecc4ac5d3952ee59b5fd47e53d45718.
Reverted https://github.com/pytorch/pytorch/pull/74727 on behalf of https://github.com/mehtanirav due to Breaking multiple internals builds and tests
2022-07-11 18:29:45 +00:00
23bdb570cf
Reland: Enable dim=None for torch.sum ( #79881 )
...
Part of #29137
Reland of #75845
Pull Request resolved: https://github.com/pytorch/pytorch/pull/79881
Approved by: https://github.com/albanD , https://github.com/kulinseth
2022-07-09 00:54:42 +00:00
39f659c3ba
Revert "[Array API] Add linalg.vecdot ( #70542 )"
...
This reverts commit 74208a9c68b5892b9dde39d06350fe7b92691429.
Reverted https://github.com/pytorch/pytorch/pull/70542 on behalf of https://github.com/malfet due to Broke CUDA-10.2 for vecdot_bfloat16, see 74208a9c68
2022-07-08 22:56:51 +00:00
cc3126083e
Remove split functional wrapper ( #74727 )
...
Pull Request resolved: https://github.com/pytorch/pytorch/pull/74727
Approved by: https://github.com/albanD
2022-07-08 19:21:22 +00:00
74208a9c68
[Array API] Add linalg.vecdot ( #70542 )
...
This PR adds the function `linalg.vecdot` specified by the [Array
API](https://data-apis.org/array-api/latest/API_specification/linear_algebra_functions.html#function-vecdot )
For the complex case, it chooses to implement \sum x_i y_i. See the
discussion in https://github.com/data-apis/array-api/issues/356
Edit. When it comes to testing, this function is not quite a binopt, nor a reduction opt. As such, we're this close to be able to get the extra testing, but we don't quite make it. Now, it's such a simple op that I think we'll make it without this.
Resolves https://github.com/pytorch/pytorch/issues/18027 .
cc @mruberry @rgommers @pmeier @asmeurer @leofang @AnirudhDagar @asi1024 @emcastillo @kmaehashi
Pull Request resolved: https://github.com/pytorch/pytorch/pull/70542
Approved by: https://github.com/IvanYashchuk , https://github.com/mruberry
2022-07-08 15:37:58 +00:00
37a5819665
Make slogdet, linalg.sloget and logdet support metatensors ( #79742 )
...
This PR also adds complex support for logdet, and makes all these
functions support out= and be composite depending on one function. We
also extend the support of `logdet` to complex numbers and improve the
docs of all these functions.
We also use `linalg_lu_factor_ex` in these functions, so we remove the
synchronisation present before.
Pull Request resolved: https://github.com/pytorch/pytorch/pull/79742
Approved by: https://github.com/IvanYashchuk , https://github.com/albanD
2022-07-01 16:09:21 +00:00
04407431ff
MAINT: Harmonize argsort params with array_api ( #75162 )
...
Closes [#70922 ](https://github.com/pytorch/pytorch/issues/70922 ).
- Does what it says on the tin.
- No non-standard implementation details.
Pull Request resolved: https://github.com/pytorch/pytorch/pull/75162
Approved by: https://github.com/rgommers , https://github.com/nikitaved , https://github.com/mruberry
2022-06-09 12:32:01 +00:00
c6215b343c
Deprecate torch.lu_solve
...
**BC-breaking note**:
This PR deprecates `torch.lu_solve` in favor of `torch.linalg.lu_solve_factor`.
A upgrade guide is added to the documentation for `torch.lu_solve`.
Note this PR DOES NOT remove `torch.lu_solve`.
Pull Request resolved: https://github.com/pytorch/pytorch/pull/77637
Approved by: https://github.com/malfet
2022-06-07 22:50:14 +00:00
f7b9a46880
Deprecate torch.lu
...
**BC-breaking note**:
This PR deprecates `torch.lu` in favor of `torch.linalg.lu_factor`.
A upgrade guide is added to the documentation for `torch.lu`.
Note this PR DOES NOT remove `torch.lu`.
Pull Request resolved: https://github.com/pytorch/pytorch/pull/77636
Approved by: https://github.com/malfet
2022-06-07 22:50:14 +00:00
f091b3fb4b
Update torch.lu_unpack docs
...
As per title
Pull Request resolved: https://github.com/pytorch/pytorch/pull/77635
Approved by: https://github.com/malfet
2022-06-07 22:50:13 +00:00
bd08d085b0
Update argmin docs to reflect the code behavior ( #78888 )
...
Fixes #78791
Pull Request resolved: https://github.com/pytorch/pytorch/pull/78888
Approved by: https://github.com/ezyang
2022-06-07 00:44:39 +00:00
c461d8a977
[primTorch] refs: hsplit, vsplit ( #78418 )
...
As per title
TODO:
* [x] Add error inputs (already exist)
Pull Request resolved: https://github.com/pytorch/pytorch/pull/78418
Approved by: https://github.com/mruberry
2022-06-06 19:54:05 +00:00
57c117d556
update signbit docs and add -0. to reference testing for unary and binary functions. ( #78349 )
...
Fixes https://github.com/pytorch/pytorch/issues/53963
Pull Request resolved: https://github.com/pytorch/pytorch/pull/78349
Approved by: https://github.com/mruberry
2022-06-06 13:48:08 +00:00
416f581eb1
Updating torch.log example
...
Fixes issue #78301
Pull Request resolved: https://github.com/pytorch/pytorch/pull/78776
Approved by: https://github.com/ngimel
2022-06-03 00:57:35 +00:00
388d44314d
Fix docs for torch.real ( #78644 )
...
Non-complex types are supported
```python
>>> import torch
>>> z = torch.zeros(5)
>>> torch.real(z.float())
tensor([0., 0., 0., 0., 0.])
>>> torch.real(z.int())
tensor([0, 0, 0, 0, 0], dtype=torch.int32)
```
Pull Request resolved: https://github.com/pytorch/pytorch/pull/78644
Approved by: https://github.com/mruberry , https://github.com/anjali411
2022-06-02 04:17:03 +00:00
