Strided masked reduction: mean (2nd try)#67088
Strided masked reduction: mean (2nd try)#67088pearu wants to merge 10 commits intogh/pearu/9/basefrom
Conversation
[ghstack-poisoned]
CI Flow Status⚛️ CI FlowRuleset - Version:
You can add a comment to the PR and tag @pytorchbot with the following commands: # ciflow rerun, "ciflow/default" will always be added automatically
@pytorchbot ciflow rerun
# ciflow rerun with additional labels "-l <ciflow/label_name>", which is equivalent to adding these labels manually and trigger the rerun
@pytorchbot ciflow rerun -l ciflow/scheduled -l ciflow/slowFor more information, please take a look at the CI Flow Wiki. |
🔗 Helpful links
💊 CI failures summary and remediationsAs of commit 171dd33 (more details on the Dr. CI page): 💚 💚 Looks good so far! There are no failures yet. 💚 💚 This comment was automatically generated by Dr. CI (expand for details).Please report bugs/suggestions to the (internal) Dr. CI Users group. |
[ghstack-poisoned]
[ghstack-poisoned]
[ghstack-poisoned]
[ghstack-poisoned]
|
I can reproduce the fweights.Using low=1 fixes the issue... |
[ghstack-poisoned]
[ghstack-poisoned]
|
@cpuhrsch has imported this pull request. If you are a Facebook employee, you can view this diff on Phabricator. |
Summary: Pull Request resolved: #67088 Test Plan: Imported from OSS Reviewed By: pbelevich Differential Revision: D31914868 Pulled By: cpuhrsch fbshipit-source-id: beda9d32ea65bcae31c2c0181f95ad23c6631075
|
Unlanded, broke windows builds, e.g. https://github.com/pytorch/pytorch/runs/4014552209?check_suite_focus=true |
|
This pull request has been reverted by 273ab55. To re-land this change, follow these steps. |
|
This pull request has been reverted by 526053706411ec3f2f8024849d951d6bf37b53a4. To re-land this change, follow these steps. |
|
The failure is likely related to numpy/numpy#18552 which has a fix available in numpy 1.20.2. |
|
TIL about np.mean with |
Differential Revision: [D31914868](https://our.internmc.facebook.com/intern/diff/D31914868) [ghstack-poisoned]
Differential Revision: [D31914868](https://our.internmc.facebook.com/intern/diff/D31914868) [ghstack-poisoned]
|
@cpuhrsch could you try importing this PR? |
|
@ngimel NumPy defines the input to I think that functionality is covered by the masked reduction RFC, but I could see us potentially wanting to add this for NumPy compatibility? There is also some discussion around using where in the RFC itself, but only to decide on the definition of the mask semantics (i.e. does |
|
Just for completeness note that the definition of mask differs between NumPy's MaskedArray and the where argument. |
|
@cpuhrsch I agree that |
|
@ngimel - Ideally we're pythonic in the sense of "one obvious way to do it", but I can also see that there's value in adding this feature for compatibility. I do worry though that we'll end up in a situation where people are doing the usual "masked out row" or "fully masked out" inputs and then run into gradient issues, which we're currently working on resolving via MaskedTensor. |
|
@pearu - could you rebase? |
Differential Revision: [D31914868](https://our.internmc.facebook.com/intern/diff/D31914868) [ghstack-poisoned]
|
@cpuhrsch rebase done. |
|
@anjali411 has imported this pull request. If you are a Facebook employee, you can view this diff on Phabricator. |
Summary: Pull Request resolved: #67088 Stack from [ghstack](https://github.com/ezyang/ghstack): * __->__ #67088 Test Plan: Imported from OSS Reviewed By: anjali411 Differential Revision: D32070264 Pulled By: cpuhrsch fbshipit-source-id: 08a91550dd24fb0f51abf06591a0e26186c4f9f9
Stack from ghstack:
Differential Revision: D32070264