Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

[MPS] add aten::normal.Tensor_float aten::normal.float_Tensor aten::normal.Tensor_Tensor #80297

Closed
wants to merge 1 commit into from

Conversation

qqaatw
Copy link
Collaborator

@qqaatw qqaatw commented Jun 26, 2022

Fixes #ISSUE_NUMBER

@facebook-github-bot
Copy link
Contributor

facebook-github-bot commented Jun 26, 2022

🔗 Helpful links

❌ 1 New Failures

As of commit a62cefd (more details on the Dr. CI page):

Expand to see more
  • 1/1 failures introduced in this PR

🕵️ 1 new failure recognized by patterns

The following CI failures do not appear to be due to upstream breakages

See GitHub Actions build trunk / macos-11-py3-x86-64 / test (default, 2, 2, macos-12) (1/1)

Step: "Test" (full log | diagnosis details | 🔁 rerun)

2022-06-27T23:18:19.5754780Z FAIL [6.145s]: tes...zation.core.test_workflow_ops.TestFakeQuantizeOps)
2022-06-27T23:18:19.5190630Z   test_subgraph_rewriter_traced_as_callable (quantization.fx.test_subgraph_rewriter.TestSubgraphRewriter) ... ok (0.028s)
2022-06-27T23:18:19.5382380Z   test_subgraph_rewriter_with_oneliner_pattern (quantization.fx.test_subgraph_rewriter.TestSubgraphRewriter) ... ok (0.019s)
2022-06-27T23:18:19.5575260Z   test_subgraph_writer_replace_consecutive_submodules (quantization.fx.test_subgraph_rewriter.TestSubgraphRewriter) ... ok (0.019s)
2022-06-27T23:18:19.5604680Z   test_get_fqn_to_example_inputs_complex_args (quantization.core.test_utils.TestUtils)
2022-06-27T23:18:19.5635840Z Test that we can record complex example inputs such as lists and dicts ... ok (0.006s)
2022-06-27T23:18:19.5664710Z   test_get_fqn_to_example_inputs_default_kwargs (quantization.core.test_utils.TestUtils)
2022-06-27T23:18:19.5697230Z Test that we can get example inputs for functions with default keyword arguments ... ok (0.006s)
2022-06-27T23:18:19.5753540Z   test_get_fqn_to_example_inputs_simple (quantization.core.test_utils.TestUtils) ... ok (0.006s)
2022-06-27T23:18:19.5754150Z 
2022-06-27T23:18:19.5754350Z ======================================================================
2022-06-27T23:18:19.5754780Z FAIL [6.145s]: test_fq_module_per_tensor (quantization.core.test_workflow_ops.TestFakeQuantizeOps)
2022-06-27T23:18:19.5755560Z ----------------------------------------------------------------------
2022-06-27T23:18:19.5755890Z Traceback (most recent call last):
2022-06-27T23:18:19.5756300Z   File "/Users/runner/work/pytorch/pytorch/test/quantization/core/test_workflow_ops.py", line 520, in test_fq_module_per_tensor
2022-06-27T23:18:19.5756720Z     X=hu.tensor(shapes=hu.array_shapes(1, 5,),
2022-06-27T23:18:19.5757480Z   File "/Users/runner/miniconda3/envs/build/lib/python3.8/site-packages/hypothesis/core.py", line 1248, in wrapped_test
2022-06-27T23:18:19.5757960Z     raise the_error_hypothesis_found
2022-06-27T23:18:19.5758370Z   File "/Users/runner/work/pytorch/pytorch/test/quantization/core/test_workflow_ops.py", line 536, in test_fq_module_per_tensor
2022-06-27T23:18:19.5758880Z     np.testing.assert_allclose(Y.cpu().detach().numpy(), Y_prime.cpu().detach().numpy(), rtol=tolerance, atol=tolerance)
2022-06-27T23:18:19.5759580Z   File "/Users/runner/miniconda3/envs/build/lib/python3.8/site-packages/numpy/testing/_private/utils.py", line 1532, in assert_allclose
2022-06-27T23:18:19.5760040Z     assert_array_compare(compare, actual, desired, err_msg=str(err_msg),

This comment was automatically generated by Dr. CI (expand for details).

Please report bugs/suggestions to the (internal) Dr. CI Users group.

Click here to manually regenerate this comment.

@mruberry mruberry added the triaged This issue has been looked at a team member, and triaged and prioritized into an appropriate module label Jun 27, 2022
@mruberry mruberry requested a review from albanD June 27, 2022 13:26
Copy link
Collaborator

@albanD albanD left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

SGTM

@albanD albanD added the ciflow/trunk Trigger trunk jobs on your pull request label Jun 27, 2022
@albanD
Copy link
Collaborator

albanD commented Jun 27, 2022

@pytorchbot merge -g

@pytorchmergebot
Copy link
Collaborator

@pytorchbot successfully started a merge job. Check the current status here

@pytorchmergebot
Copy link
Collaborator

Merge failed due to 2 additional jobs have failed, first few of them are: trunk ,macos-11-py3-x86-64 / test (default, 2, 2, macos-12)
Raised by https://github.com/pytorch/pytorch/actions/runs/2571886138

Copy link
Collaborator

@kulinseth kulinseth left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

LGTM.

@albanD
Copy link
Collaborator

albanD commented Jun 28, 2022

@pytorchbot merge -f

@pytorchmergebot
Copy link
Collaborator

@pytorchbot successfully started a merge job. Check the current status here

@pytorchmergebot
Copy link
Collaborator

@qqaatw your PR has been successfully merged.

@github-actions
Copy link

Hey @qqaatw.
You've committed this PR, but it does not have both a 'release notes: ...' and 'topics: ...' label. Please add one of each to the PR. The 'release notes: ...' label should represent the part of PyTorch that this PR changes (fx, autograd, distributed, etc) and the 'topics: ...' label should represent the kind of PR it is (not user facing, new feature, bug fix, perf improvement, etc). The list of valid labels can be found here for the 'release notes: ...' and here for the 'topics: ...'.
For changes that are 'topic: not user facing' there is no need for a release notes label.

facebook-github-bot pushed a commit that referenced this pull request Jun 30, 2022
…en::normal.Tensor_Tensor` (#80297) (#80297)

Summary:
Fixes #ISSUE_NUMBER

Pull Request resolved: #80297
Approved by: https://github.com/albanD, https://github.com/kulinseth

Test Plan: contbuild & OSS CI, see https://hud.pytorch.org/commit/pytorch/pytorch/e1b15b7a047938ca0ef4e8808a2419b882cc2dd4

Reviewed By: b0noI

Differential Revision: D37509779

fbshipit-source-id: c04365062c206e7839c45b3a109f35c7698d7774
kulinseth pushed a commit to kulinseth/pytorch that referenced this pull request Jul 9, 2022
@qqaatw
Copy link
Collaborator Author

qqaatw commented Jul 15, 2022

@pytorchbot label "release notes: mps" "topic: new feature"

@pytorch-bot pytorch-bot bot added release notes: mps Release notes category topic: new features topic category labels Jul 15, 2022
atalman pushed a commit to atalman/pytorch that referenced this pull request Jul 22, 2022
atalman added a commit that referenced this pull request Jul 25, 2022
* MPS: Fixes (#78930)

Cast integer to float in UnaryOps
Add tensor dtype in key generation
Enable FP16 scalars and use placeholder for alpha tensor in add/sum ops

Fixes #ISSUE_NUMBER

Pull Request resolved: #78930
Approved by: https://github.com/albanD

* MPS: Binary cast fix by proper type promotion and remove spurious copy warning (#79185)

Fixes #78019, #78020
Fixes #79185
Pull Request resolved: #79185
Approved by: https://github.com/albanD, https://github.com/razarmehr

* MPS: add exponential op (#79188)

Add exponential distribution

Fixes #ISSUE_NUMBER

Pull Request resolved: #79188
Approved by: https://github.com/razarmehr, https://github.com/albanD

* [MPS] Delete unused vars from OperationUtils.mm

Pull Request resolved: #79514

Approved by: https://github.com/kulinseth, https://github.com/albanD

* [MPS] Fix getDefaultGenerator and copy_kernel_mps

Returning reference to stack memory is really bad

Pull Request resolved: #79515

Approved by: https://github.com/albanD

* [MPS][BE]Do not use `new/delete[]` in `chainViewOperation`

`std::array` will do just fine

Pull Request resolved: #79516

Approved by: https://github.com/albanD

* [MPS] Support stride of stride

Fixes #79181

Pull Request resolved: #79521

Approved by: https://github.com/kulinseth

* MPS: TopK raise an error if K>16 (#79677)

* Error out in TopK when k>16.
* Add a test case too.

Fixes #78915

Pull Request resolved: #79677
Approved by: https://github.com/albanD

* [MPS]: Add fix for squeezed input axes handling in BCE loss (#79676)

Fixes #79527

Pull Request resolved: #79676
Approved by: https://github.com/razarmehr, https://github.com/albanD

* MPS: Add amax and amin Ops with tests  (#79682)

* Add amax and amin with tests

Fixes #ISSUE_NUMBER

Pull Request resolved: #79682
Approved by: https://github.com/albanD

* [MPS] Fix torch.uint8 support (#80049)

`ScalarType.Byte` should be cast to `MPSDataTypeUInt8`
And support for `torch.int8` as well as test those conversions in `TestMPS.test_to`

Fixes #80006

Pull Request resolved: #80049
Approved by: https://github.com/albanD

* [MPS] Fix binary ops between int32 tensor with int64 scalar (#80220)

For some reason, tensor *op* scalar does not follow the normal binary promotion rules
So cast output tensor to expected type if needed
It seems that one should have casted input tensors to expected output tensor type, but it does not really work for boolean binary ops, so...
Add output tensor type/shape to cached graph key
Extend `TestMPS. test_add_scalars` to test for this regression

Fixes #79835

Pull Request resolved: #80220
Approved by: https://github.com/albanD

* [MPS] Add equal operator (#80195)

Which is, in essence is composite of `eq`->`all`->`item`
`native/mps/operators/Equal.cpp` is an almost verbatim copy of `native/cuda/Equal.cpp`

Fix codegen by generating MPSFunctions headers

Pull Request resolved: #80195
Approved by: https://github.com/albanD

* [MPS] add `aten::normal.Tensor_float` `aten::normal.float_Tensor` `aten::normal.Tensor_Tensor` (#80297)

Fixes #ISSUE_NUMBER

Pull Request resolved: #80297
Approved by: https://github.com/albanD, https://github.com/kulinseth

* [MPS] Add flip (#80214)

Fixes #ISSUE_NUMBER

Pull Request resolved: #80214
Approved by: https://github.com/DenisVieriu97, https://github.com/albanD

* [MPS] Add logical ops (#80216)

This PR adds `logical_not`, `logical_and`, `logical_or`, `logical_xor`.
Pull Request resolved: #80216
Approved by: https://github.com/albanD, https://github.com/kulinseth

* [MPS] Add glu (#79866)

Adds mps op for `aten::glu.out`.

Pull Request resolved: #79866
Approved by: https://github.com/kulinseth, https://github.com/albanD

* [MPS] Fix std/var cache issue (#80502)

Use `getTensorsStringKey` which has tensor shape info added as part of the key to prevent cache lookup issue when the shape of input tensor is changed.

Fixes #80499

Pull Request resolved: #80502
Approved by: https://github.com/malfet, https://github.com/kulinseth

* Add scatter support for view operations (#79939)

* Add scatter support for view operations; #78074, #78886, #79672
* Update test_slicing_replace_column to properly test different sizes
* Handle in-place changes for binary ops; add new testcase
* Add new view ops testing scatter; add MPSDebugConfig.h config file for debugging purposes
* Merge gatherViewTensor and scatterViewTensor into a generic function
* Add scatter on demand in scatterViewOperation instead of caching it into a generic graph
* Create separate graphs for scatter and gather;
* Create scatter graph at scatter time

Fixes #ISSUE_NUMBER

Pull Request resolved: #79939
Approved by: https://github.com/razarmehr

* MPS: Fix handling of 1D tensors in linear backward (#80759)

Fixes ##79784

Pull Request resolved: #80759
Approved by: https://github.com/ezyang

* [MPS] Move the View ops to a separate file and reduce the number of graphs created (#80491)

This is dependent on the PR to go in first: #79939

Remove the data_ptr from the View Graph key which reduces the number of
graphs created significantly.

Don't wait when copying from MPS to MPS tensors

Pull Request resolved: #80491
Approved by: https://github.com/malfet

* [MPS] Add softplus backward (#79873)

Fixes #ISSUE_NUMBER

Pull Request resolved: #79873
Approved by: https://github.com/malfet

* [MPS] Add argmin (#80828)

This PR

1. adds argmin
2. refactors `reduction_type` in `ReduceOps.mm` with enum.

Co-authored by Kulin Seth <kulinseth@gmail.com>
Pull Request resolved: #80828
Approved by: https://github.com/malfet

* [MPS] Fix LSTM batch_first output transposed (#80597)

The output of LSTM with `batch_first` should be transposed back to batch first format.

Fixes #80306

Pull Request resolved: #80597
Approved by: https://github.com/kulinseth

* [MPS][BE] Introduce MPSUnaryCachedGraph (#81033)

I.e. CachedGraph that has input and output tensors
Also, add `MPSGraphCache::LookUpAs` template, which combines LookUp with
static_cast to target type

Pull Request resolved: #81033
Approved by: https://github.com/kulinseth

* [MPS] Add test consistency from OpInfo based tests from PR 78504 (#79532)

Pull Request resolved: #79532
Approved by: https://github.com/albanD, https://github.com/malfet

* [MPS] Add huber loss (#80163)

Fixes #ISSUE_NUMBER

Pull Request resolved: #80163
Approved by: https://github.com/kulinseth, https://github.com/malfet

* Remove two tests dependent on the MPS serialization checkin.

* Fix lint error (FLAKE8) F401

* Remove the serialization test from test_mps as its support is not there in 1.12.1.

Co-authored-by: Kulin Seth <kulinseth@gmail.com>
Co-authored-by: Nikita Shulga <nikita.shulga@gmail.com>
Co-authored-by: Kulin Seth <kulin_seth@apple.com>
Co-authored-by: Abhishek Pathak <abhipathak97@gmail.com>
Co-authored-by: Nikita Shulga <nshulga@fb.com>
Co-authored-by: qqaatw <qqaatw@gmail.com>
Co-authored-by: Ramin Azarmehr <razarmehr@apple.com>
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
ciflow/trunk Trigger trunk jobs on your pull request cla signed Merged open source release notes: mps Release notes category topic: new features topic category triaged This issue has been looked at a team member, and triaged and prioritized into an appropriate module
Projects
None yet
Development

Successfully merging this pull request may close these issues.

None yet

7 participants