Commit Graph

1318 Commits

Author SHA1 Message Date
yanbing-j
2edcc9e600 Update ideep for ideep conv changes -v2 (#79258)
The original PR https://github.com/pytorch/pytorch/pull/78238 has been covered. This is to reopen.
Pull Request resolved: https://github.com/pytorch/pytorch/pull/79258
Approved by: https://github.com/frank-wei
2022-06-11 14:19:03 +00:00
Min Si
8d9e5d0b5f [pytorch] Update Gloo submodule to include rocm fixes (#77207)
Summary: Update gloo submodule commit to include recent rocm-related fixes in gloo.

Test Plan: OSS pytorch CI/CD

Differential Revision: D36298978

Pull Request resolved: https://github.com/pytorch/pytorch/pull/77207
Approved by: https://github.com/kwen2501
2022-06-08 01:49:07 +00:00
Jon Janzen
e806d13214 [Privacy][Codemod][TransferMegarepoThirdPartyLibraryOwnership] Asset Ownership Update For asset://code.third_party_library/fbsource/xplat%2Fcaffe2%2Fthird_party (#78937)
Differential Revision: D36817082

Pull Request resolved: https://github.com/pytorch/pytorch/pull/78937
Approved by: https://github.com/dagitses
2022-06-07 00:16:44 +00:00
Sergei Vorobev
a0a23c6ef8 [bazel] make it possible to build the whole world, update CI (#78870)
Fixes https://github.com/pytorch/pytorch/issues/77509

This PR supersedes https://github.com/pytorch/pytorch/pull/77510.
It allows both `bazel query //...` and `bazel build --config=gpu //...` to work.

Concretely the changes are:
1. Add "GenerateAten" mnemonic -- this is a convenience thing, so anybody who uses [Remote Execution](https://bazel.build/docs/remote-execution) can add a

```
build:rbe --strategy=GenerateAten=sandboxed,local
```

line to the `~/.bazelrc` and build this action locally (it doesn't have hermetic dependencies at the moment).

2. Replaced few `http_archive` repos by the proper existing submodules to avoid code drift.
3. Updated `pybind11_bazel` and added `python_version="3"` to `python_configure`. This prevents hard-to-debug error that are caused by an attempt to build with python2 on the systems where it's a default python (Ubuntu 18.04 for example).
4. Added `unused_` repos, they purpose is to hide the unwanted submodules of submodules that often have bazel targets in them.
5. Updated CI to build //... -- this is a great step forward to prevent regressions in targets not only in the top-level BUILD.bazel file, but in other folders too.
6. Switch default bazel build to use gpu support.
Pull Request resolved: https://github.com/pytorch/pytorch/pull/78870
Approved by: https://github.com/ezyang
2022-06-06 21:58:47 +00:00
Jon Janzen
03847808a0 Add all bzl files per D36874458 2022-06-06 09:40:19 -07:00
Linbin Yu
1683a2618d rename BUILD.buck to BUCK.oss (#78792)
rename BUILD.buck to BUCK.oss to better reflect that it's the OSS version of BUCK build, not the one shared with Bazel
Pull Request resolved: https://github.com/pytorch/pytorch/pull/78792
Approved by: https://github.com/kit1980
2022-06-03 07:23:16 +00:00
Edward Z. Yang
eee2aa14a6 Register std_mean ref as a decomposition
Signed-off-by: Edward Z. Yang <ezyangfb.com>

Pull Request resolved: https://github.com/pytorch/pytorch/pull/78468

Approved by: https://github.com/ngimel
2022-05-31 18:59:16 +00:00
yanbing-j
f09f6aadb4 Update ideep for ideep conv changes (#78238)
Pull Request resolved: https://github.com/pytorch/pytorch/pull/78238
Approved by: https://github.com/dagitses
2022-05-31 14:18:37 +00:00
Nikita Shulga
472d67a727 Revert "move XNNPACK buck build to shared build file (#77941)"
This reverts commit b8b46f932b.

This change were reverted internally but has not been populated to
neither `master` nor `fbsync` branches yet
2022-05-30 10:59:51 -07:00
Linbin Yu
b8b46f932b move XNNPACK buck build to shared build file (#77941)
Summary:
This diff moved the XNNPACK buck build to a shared build file in xplat/caffe2/third_party, so it can be reused by OSS buck CI in the future. There's no functionality change.

**Background**: as we are moving to github-first, we want community to receive more signals from our internal build. XNNPACK is part of pytorch mobile build so we want to add it to OSS BUCK CI.

**How it works**: all XNNPACK targets are defined in xplat/caffe2/third_party/xnnpack_defs.bzl. When we build it internally, the XNNPACK source is still at xplat/third-party/XNNPACK and we will load that bzl file in xplat/third-party/XNNPACK/BUCK. Everything should work as before. In OSS build, XNNPACK is a submodule in xplat/caffe2/third_party and we will load the same bzl file in pytorch/third_party/BUILD.buck.

**Wrapper Generation**: the wrapper generation script is moved to xplat/caffe2/third_party/generate-xnnpack-wrappers.py. It will take an optional argument for the path of XNNPACK (they are different in internal build and OSS build). The wrapper files will always be generated at the parent folder of XNNPACK source. But the src_defs.bzl and wrapper_defs.bzl will always be in xplat/caffe2/third_party/ (they are now called xnnpack_src_defs.bzl and xnnpack_wrapper_defs.bzl). For OSS build this script will only be used in CI, and the generated files will not be committed.

**Next Steps:** Once landed, I will try to build XNNPACK in OSS BUCK using xnnpack_defs.bzl. Meta-specific symbols need to be resolved, so there will be some refactors to the build file.

Test Plan: buck build xplat/third-party/XNNPACK:XNNPACK

Differential Revision: D36529332

Pull Request resolved: https://github.com/pytorch/pytorch/pull/77941
Approved by: https://github.com/malfet, https://github.com/seemethere
2022-05-28 04:39:37 +00:00
Nikita Shulga
fa7117c64a Update PeachPy submodule (#78326)
Forked the repo, merged latest changes into pre-generated branch and
update pregenerared opcodes

Re-enabled NNPACK builds on MacOS

Picking f8ef1a3c0a  fixes https://github.com/pytorch/pytorch/issues/76094

Pull Request resolved: https://github.com/pytorch/pytorch/pull/78326
Approved by: https://github.com/atalman, https://github.com/albanD
2022-05-26 13:58:36 +00:00
Facebook Community Bot
08dafb761e Automated submodule update: kineto (#75687)
This is an automated pull request to update the first-party submodule for [pytorch/kineto](https://github.com/pytorch/kineto).

New submodule commit: 097622eb86

Test Plan: Ensure that CI jobs succeed on GitHub before landing.
Pull Request resolved: https://github.com/pytorch/pytorch/pull/75687
Approved by: https://github.com/robieta
2022-05-24 20:39:22 +00:00
Eddie Yan
14ab3ff484 [cuDNN V8 API] Enable cuDNN v8 API by default (#75466)
Testing via CI
Pull Request resolved: https://github.com/pytorch/pytorch/pull/75466
Approved by: https://github.com/ngimel
2022-05-17 21:54:17 +00:00
Linbin Yu
1f8049566f Re-land BUCK build for pytorch mobile (#77612)
see https://github.com/pytorch/pytorch/pull/76480
fixed most lint errors
Pull Request resolved: https://github.com/pytorch/pytorch/pull/77612
Approved by: https://github.com/kit1980
2022-05-17 00:30:13 +00:00
PyTorch MergeBot
530481ed69 Revert "[mobile] add buck build for mobile targets (#76480)"
This reverts commit 168dc70faf.

Reverted https://github.com/pytorch/pytorch/pull/76480 on behalf of https://github.com/atalman
2022-05-16 16:14:17 +00:00
Linbin Yu
168dc70faf [mobile] add buck build for mobile targets (#76480)
Create buck targets to replicate internal BUCK build, including
- XNNPACK
- QNNPACK
- C10
- aten_cpu
- torch_mobile_core
- torch_mobile_all_ops
- ptmobile_benchmark

And able to run mobilenet v2 using ptmobile_benchmark (with all ops).
Pull Request resolved: https://github.com/pytorch/pytorch/pull/76480
Approved by: https://github.com/seemethere, https://github.com/dreiss
2022-05-15 18:42:41 +00:00
Nikita Shulga
8473173c36 Remove breakpad dependency
This functionality does not seem to be used
and there are some requests to update dependency.

Add `third_party` to torch_cpu include directories if compiling with
Caffe2 support, as `caffe2/quantization/server/conv_dnnlowp_op.cc` depends on `third_party/fbgemm/src/RefImplementations.h`

Pull Request resolved: https://github.com/pytorch/pytorch/pull/75394
Approved by: https://github.com/janeyx99, https://github.com/seemethere
2022-05-03 20:21:55 +00:00
yanbing-j
58d773ad29 Upgrade oneDNN to v2.6.0 (#75398)
Summary:
This PR upgrades oneDNN to v2.6.0.

v2.6.0 changes:

- Improved performance for future Intel Xeon® Scalable processors (code name Sapphire Rapids). The functionality requires Linux kernel 5.16 or later.
- Improved performance of matmul primitive for processors with Intel AVX-512 support.
- Introduced bfloat16 destination support for int8 convolution, matmul and inner product primitives for processors with Intel AVX-512 support and or future Intel Xeon® Scalable processors (code name Sapphire Rapids)
- Extended RNN primitive with support for [AUGRU cell](https://oneapi-src.github.io/oneDNN/dev_guide_rnn.html#augru).
- Added support for non-zero negative slope in ReLU post-op for batch normalization primitive.
- Introduced support for mixed source and destination data types in softmax primitive.
- Introduced [persistent cache API](https://oneapi-src.github.io/oneDNN/dev_guide_persistent_cache.html). This functionality allows to serialize and reuse JIT kernels.

Pull Request resolved: https://github.com/pytorch/pytorch/pull/75398

Reviewed By: dagitses, bigfootjon

Differential Revision: D35976827

Pulled By: frank-wei

fbshipit-source-id: a77ae15cd77fc7c114ab9722453c28dc64aac679
(cherry picked from commit e376698d3c772aaa2dfbe51a4d1a75c8d17d0eee)
2022-05-02 22:07:42 +00:00
Eli Uriegas
a9a17ce626 third_party: Fix build_bundled script
This script was hanging up on a couple of licenses not being parsed
correctly / not accounted for in the original pass through

Should resolve issues with tests not running 100% correctly

Signed-off-by: Eli Uriegas <eliuriegasfb.com>

Pull Request resolved: https://github.com/pytorch/pytorch/pull/76336

Signed-off-by: Eli Uriegas <eliuriegas@fb.com>

Approved by: https://github.com/ezyang
2022-04-25 21:06:03 +00:00
Sergii Dymchenko
e5282c3cb8 Again add first version of Buck build workflow
Next attempt after https://github.com/pytorch/pytorch/pull/75815 (that PR was reverted for breaking internal build because of BUCK.github file name doesn't get synced internal->github).

Pull Request resolved: https://github.com/pytorch/pytorch/pull/76156
Approved by: https://github.com/seemethere
2022-04-21 16:26:10 +00:00
Jiyuan Zhang
2602a5e76f fix local_thread:pytorch embeddingbag
Test Plan: buck test //deeplearning/fbgemm:EmbeddingSpMDMTest

Differential Revision: D35595457

Pull Request resolved: https://github.com/pytorch/pytorch/pull/75702
Approved by: https://github.com/jianyuh
2022-04-18 19:52:04 +00:00
PyTorch MergeBot
b5a25180f1 Revert "Add first version of Buck build workflow"
This reverts commit de66304aa5.

Reverted https://github.com/pytorch/pytorch/pull/75815 on behalf of https://github.com/malfet
2022-04-17 18:48:44 +00:00
PyTorch MergeBot
d79d9fa283 Revert "Remove breakpad dependency"
This reverts commit 9aa3c7fd83.

Reverted https://github.com/pytorch/pytorch/pull/75394 on behalf of https://github.com/malfet
2022-04-17 17:58:51 +00:00
Nikita Shulga
9aa3c7fd83 Remove breakpad dependency
This functionality does not seem to be used
and there are some requests to update dependency

Pull Request resolved: https://github.com/pytorch/pytorch/pull/75394
Approved by: https://github.com/janeyx99, https://github.com/seemethere
2022-04-17 17:43:45 +00:00
Sergii Dymchenko
de66304aa5 Add first version of Buck build workflow
Next attempt after https://github.com/pytorch/pytorch/pull/75360 (that PR was reverted for breaking internal build because of BUCK file names conflicts).
Pull Request resolved: https://github.com/pytorch/pytorch/pull/75815
Approved by: https://github.com/linbinyu
2022-04-14 20:18:13 +00:00
PyTorch MergeBot
496d4bb7ca Revert "Add first version of Buck build workflow"
This reverts commit 045228bad1.

Reverted https://github.com/pytorch/pytorch/pull/75360 on behalf of https://github.com/kit1980
2022-04-14 18:01:33 +00:00
Sergii Dymchenko
045228bad1 Add first version of Buck build workflow
Pull Request resolved: https://github.com/pytorch/pytorch/pull/75360
Approved by: https://github.com/linbinyu, https://github.com/seemethere
2022-04-13 23:50:23 +00:00
dzdang
25aa251f37 updated cudnn frontend to v0.6.1
cudnn frontend v0.6.0 had an issue. we upgrade to cudnn frontend to v0.6.1. see reverted PR: https://github.com/pytorch/pytorch/pull/75481
Pull Request resolved: https://github.com/pytorch/pytorch/pull/75615
Approved by: https://github.com/jerryzh168
2022-04-12 02:15:04 +00:00
David Dang
771527fc6b Revert D35489740: [pytorch][PR] Updated cudnn_frontend submodule to v0.6
Test Plan: revert-hammer

Differential Revision:
D35489740 (6e11435c41)

Original commit changeset: b24a71e6774f

Original Phabricator Diff: D35489740 (6e11435c41)

fbshipit-source-id: eb2f81c418a9e9cfb14dbf01f6d993f142474f75
(cherry picked from commit c03905bc5a61cb04033bef3c1ebb90cfb332f33f)
2022-04-08 22:15:22 +00:00
dzdang
6e11435c41 Updated cudnn_frontend submodule to v0.6 (#75481)
Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/75481

Reviewed By: jerryzh168

Differential Revision: D35489740

Pulled By: dzdang

fbshipit-source-id: b24a71e6774f54e0035d8e3e4b91290b17ba3330
(cherry picked from commit ea2ca3f711caf12160132b3fbb6eef4bc6d529f1)
2022-04-08 12:50:50 +00:00
acxz
f98881b1bf update eigen submodule to latest release (3.4.0) with rocm fixes
Fixes #73177

Pull Request resolved: https://github.com/pytorch/pytorch/pull/73178
Approved by: https://github.com/jeffdaily, https://github.com/suo, https://github.com/malfet
2022-04-07 18:54:58 +00:00
Brian Coutinho
899326ad5e [kineto] submodule update and fixes
Kineto submodule update and fixes
Pull Request resolved: https://github.com/pytorch/pytorch/pull/75206
Approved by: https://github.com/aaronenyeshi, https://github.com/malfet
2022-04-06 22:11:24 +00:00
Facebook Community Bot
d0adb5ff26 Automated submodule update: FBGEMM (#74633)
Summary:
This is an automated pull request to update the first-party submodule for [pytorch/FBGEMM](https://github.com/pytorch/FBGEMM).

New submodule commit: ef22aabc8b

Pull Request resolved: https://github.com/pytorch/pytorch/pull/74633

Test Plan: Ensure that CI jobs succeed on GitHub before landing.

Reviewed By: jianyuh

Differential Revision: D35088395

Pulled By: geyyer

fbshipit-source-id: cb6808719a545c318302ed2770b3c7fa459fe169
(cherry picked from commit dfba6a9441136ff8563e80e6a09c555ad6a3af5a)
2022-03-24 19:09:57 +00:00
Facebook Community Bot
f167a3f95b Automated submodule update: FBGEMM (#74447)
Summary:
This is an automated pull request to update the first-party submodule for [pytorch/FBGEMM](https://github.com/pytorch/FBGEMM).

New submodule commit: d7bc9d8f4a

Pull Request resolved: https://github.com/pytorch/pytorch/pull/74447

Test Plan: Ensure that CI jobs succeed on GitHub before landing.

Reviewed By: jspark1105

Differential Revision: D35002422

fbshipit-source-id: 5376d122d1af0b4ee99ff692fe91983273eb4e61
(cherry picked from commit 95650ebd1904f0131685c689b348bf76fa614781)
2022-03-23 04:52:03 +00:00
Facebook Community Bot
38512d90c1 Automated submodule update: FBGEMM (#74423)
Summary:
This is an automated pull request to update the first-party submodule for [pytorch/FBGEMM](https://github.com/pytorch/FBGEMM).

New submodule commit: 43c0f12abd

Pull Request resolved: https://github.com/pytorch/pytorch/pull/74423

Test Plan: Ensure that CI jobs succeed on GitHub before landing.

Reviewed By: jspark1105

Differential Revision: D34993535

fbshipit-source-id: c40e2100810b40b58a2b3f536e5b132fc4b8acae
(cherry picked from commit 65e17825f4ce975417afbc26833d20eaac25ea8f)
2022-03-19 14:43:07 +00:00
Facebook Community Bot
c064ea07f6 Automated submodule update: FBGEMM (#74409)
Summary:
This is an automated pull request to update the first-party submodule for [pytorch/FBGEMM](https://github.com/pytorch/FBGEMM).

New submodule commit: ecc98a9d3e

Pull Request resolved: https://github.com/pytorch/pytorch/pull/74409

Test Plan: Ensure that CI jobs succeed on GitHub before landing.

Reviewed By: jspark1105

Differential Revision: D34984886

fbshipit-source-id: 6992460919800ff5e17b373d348d40932fb1dcb7
(cherry picked from commit 55648f01b230e69158e0aab216311e2a9393a8f0)
2022-03-18 23:46:22 +00:00
Jon Janzen
0a0502edf6 Revert D34957139: [pytorch][PR] Automated submodule update: FBGEMM
Test Plan: revert-hammer

Differential Revision:
D34957139 (d8c0f2f35d)

Original commit changeset: 67672c6f10bb

Original Phabricator Diff: D34957139 (d8c0f2f35d)

fbshipit-source-id: c50f66809f913f80098b4acbdf7778dfd2d1fc06
(cherry picked from commit f730ccf6e1b1b678cd67a0e38df04d9b9dbf450c)
2022-03-17 20:04:49 +00:00
Facebook Community Bot
d8c0f2f35d Automated submodule update: FBGEMM (#74369)
Summary:
This is an automated pull request to update the first-party submodule for [pytorch/FBGEMM](https://github.com/pytorch/FBGEMM).

New submodule commit: bedfcaeea2

Pull Request resolved: https://github.com/pytorch/pytorch/pull/74369

Test Plan: Ensure that CI jobs succeed on GitHub before landing.

Reviewed By: malfet

Differential Revision: D34957139

fbshipit-source-id: 67672c6f10bb0f6edc48e4c3b926e356313c98f7
(cherry picked from commit de2fadf31cc0186214ff1dd99cfb1c15fedc8ecd)
2022-03-17 18:09:54 +00:00
Nikita Shulga
7906ac40dd Advance fbgemm submodule
This reverts accidentally introduced update revert by ef8995ff93
Pull Request resolved: https://github.com/pytorch/pytorch/pull/74325
Approved by: https://github.com/suo
2022-03-16 19:51:26 +00:00
Nikita Shulga
ef8995ff93
[GHA][BE] Remove unneeded shellcheck suppressions (#74308)
Also fix typo in `x86_32` binaries installation command, which results
in unintended double build (hattip to @suo for finding it)
2022-03-16 11:21:58 -07:00
Facebook Community Bot
3c837a8b05 Automated submodule update: FBGEMM (#74104)
Summary:
This is an automated pull request to update the first-party submodule for [pytorch/FBGEMM](https://github.com/pytorch/FBGEMM).

New submodule commit: 43d47445d6

Pull Request resolved: https://github.com/pytorch/pytorch/pull/74104

Test Plan: Ensure that CI jobs succeed on GitHub before landing.

Reviewed By: jasonjk-park, jianyuh

Differential Revision: D34822662

fbshipit-source-id: 18d76c84b8353c8e5c306487cac43e742ace2634
(cherry picked from commit 34a88f94b3b5690f46b8ba7bd9745e9260847941)
2022-03-16 00:40:51 +00:00
Aaron Enye Shi
11759491b9 [Kineto] Manual Submodule Update (#73858)
Summary:
Pull Request resolved: https://github.com/pytorch/pytorch/pull/73858

The newer version of Kineto has changes to handle generic activities (such as RocTracer generic activities), so we can remove the older USE_KINETO_UPDATED macro and implementation of flow.linkedActivity.

This patch should bring Kineto back in sync on PyTorch CI.

Test Plan: PyTorch OSS CI needs to pass for this submodule update of third_party/kineto repo. With ciflow/cuda enabled.

Reviewed By: chaekit

Differential Revision: D34689078

Pulled By: aaronenyeshi

fbshipit-source-id: 305431cbce95ecd57c23b0485e4bb921ea2c7c0a
(cherry picked from commit aef46a287f4cee92ab96650b6a0ab81648bca374)
2022-03-14 23:18:33 +00:00
Facebook Community Bot
7070fe4d15 Automated submodule update: FBGEMM (#74088)
Summary:
This is an automated pull request to update the first-party submodule for [pytorch/FBGEMM](https://github.com/pytorch/FBGEMM).

New submodule commit: b4ba8d1368

Pull Request resolved: https://github.com/pytorch/pytorch/pull/74088

Test Plan: Ensure that CI jobs succeed on GitHub before landing.

Reviewed By: jspark1105

Differential Revision: D34811416

fbshipit-source-id: 7c47df1f56c43d5b611f2f18cc98c3e669470b75
(cherry picked from commit 9614bb863dc9b4f0e95aa48a50fd214e70c13817)
2022-03-11 15:59:47 +00:00
Facebook Community Bot
ee67ba0e6a Automated submodule update: FBGEMM (#73895)
Summary:
This is an automated pull request to update the first-party submodule for [pytorch/FBGEMM](https://github.com/pytorch/FBGEMM).

New submodule commit: 814ec8cd84

Pull Request resolved: https://github.com/pytorch/pytorch/pull/73895

Test Plan: Ensure that CI jobs succeed on GitHub before landing.

Reviewed By: jspark1105

Differential Revision: D34705004

fbshipit-source-id: 09918b633bbb950a747d806d9efc26717917a811
(cherry picked from commit 750ac00ef6369a246235209697b458d80da9cd46)
2022-03-11 02:27:55 +00:00
Nikita Shulga
e7730eaeaa Revert D34230284: [pytorch][PR] [WIP][Kineto] Manual Submodule Update
Test Plan: revert-hammer

Differential Revision:
D34230284 (c088c8fe92)

Original commit changeset: cbecaf79e51a

Original Phabricator Diff: D34230284 (c088c8fe92)

fbshipit-source-id: 6976e979c3d43ccd518248ab62082538504654c3
(cherry picked from commit 9a0ad668a5ee497b4bf741937319b734575ab1ce)
2022-03-06 17:33:01 +00:00
Aaron Enye Shi
c088c8fe92 [WIP][Kineto] Manual Submodule Update (#73090)
Summary:
The newer version of Kineto has changes to handle generic activities (such as RocTracer generic activities), so we can remove the older USE_KINETO_UPDATED macro and implementation of flow.linkedActivity.

This patch should bring Kineto back in sync on PyTorch CI.

Pull Request resolved: https://github.com/pytorch/pytorch/pull/73090

Test Plan: PyTorch OSS CI needs to pass for this submodule update of third_party/kineto repo.

Pulled By:
aaronenyeshi
aaronenyeshi

Reviewed By: briancoutinho

Differential Revision: D34230284

fbshipit-source-id: cbecaf79e51ad4b6d9cda9e855d839792a1c184b
(cherry picked from commit 2242352edd8bbb8962deca4077b7622418aad50b)
2022-03-05 21:30:56 +00:00
Facebook Community Bot
d047e475f8 Automated submodule update: FBGEMM (#73814)
Summary:
This is an automated pull request to update the first-party submodule for [pytorch/FBGEMM](https://github.com/pytorch/FBGEMM).

New submodule commit: 9f2271be0d

Pull Request resolved: https://github.com/pytorch/pytorch/pull/73814

Test Plan: Ensure that CI jobs succeed on GitHub before landing.

Reviewed By: jspark1105

Differential Revision: D34658981

fbshipit-source-id: a7f119ce6b0446803f0cf1f4d0518b95319a3696
(cherry picked from commit 7e1b11cb170e508d892cb88c9b87069063b2f797)
2022-03-05 17:30:46 +00:00
Facebook Community Bot
350ecc9d1f Automated submodule update: FBGEMM (#73277)
Summary:
This is an automated pull request to update the first-party submodule for [pytorch/FBGEMM](https://github.com/pytorch/FBGEMM).

New submodule commit: d791702c9d

Pull Request resolved: https://github.com/pytorch/pytorch/pull/73277

Test Plan: Ensure that CI jobs succeed on GitHub before landing.

Reviewed By: jasonjk-park

Differential Revision: D34411702

fbshipit-source-id: 0fe15a0549f7c30bcf61712353cc7b2089ead0ee
(cherry picked from commit f75e143bda8cf96122f921c67f9d0e9aa345a149)
2022-03-04 17:35:31 +00:00
Apoorva Garg
63932edcc7 Back out "[pytorch][PR] Support dataclasses in TorchScript"
Summary:
Original commit changeset: f5a792555c88

Original Phabricator Diff: D34398107 (d00de0d435)

Backing out as this broke fluent2 tests

Test Plan: sandcastle

Reviewed By: qihqi

Differential Revision: D34597363

fbshipit-source-id: 26bbe64b981aeb53b901cda61557614d9f28700e
(cherry picked from commit f17adfed8125ef84efaf2c8923c11a751eb7fb98)
2022-03-03 14:30:54 +00:00
Eddie Yan
dab5e2a23e [cuDNN v8 API] cuDNN benchmark, convolution bwd / transposed convolution fwd, bfloat16, conv-bias-activation fusion (#60755)
Summary:
https://github.com/pytorch/pytorch/issues/58414, https://github.com/pytorch/pytorch/issues/58859, https://github.com/pytorch/pytorch/issues/58858 #58860 https://github.com/pytorch/pytorch/issues/58861

We're currently testing performance with both "find" and "get" with this PR.

CC zasdfgbnm ptrblck ngimel puririshi98

In addition to the `USE_EXPERIMENTAL_CUDNN_V8_API` build flag, we've added a `CUDNN_V8_API_ENABLED` runtime feature flag.
`USE_EXPERIMENTAL_CUDNN_V8_API=1` will build with v8 API support while keeping all v7 functionality, with v8 usage disabled by default.
`CUDNN_V8_API_ENABLED=1` at runtime on a `USE_EXPERIMENTAL_CUDNN_V8_API=1` build uses the v8 API.
A debug flag `CUDNN_V8_API_DEBUG=1` can be used to verify which API is used when dispatching convolutions.

Note that in v7, `bfloat16` convolutions will dispatch to a native PyTorch implementation, but a fully v8 enabled build will dispatch to cuDNN implementations.

Pull Request resolved: https://github.com/pytorch/pytorch/pull/60755

Reviewed By: mruberry

Differential Revision: D34393940

Pulled By: ngimel

fbshipit-source-id: 5c317d3aad63336ea416a51a43cf8b7d27aaca21
(cherry picked from commit 3bfc549ce57cee691f83dc894ac7adb4b7882459)
2022-03-02 03:33:02 +00:00