Skip to content

Commit c0670d4

Browse files
lizhouyufacebook-github-bot
authored andcommitted
Revert TorchRec MPZCH modules to wait for FBGEMM CUDA kernel being published (#3065)
Summary: Pull Request resolved: #3065 ### Major changes - Revert changes made in D75559591 to avoid potential conflicts. - Wait for FBGEMM CUDA kernel to be fully published, then land the changes in D75559591 again. Reviewed By: aporialiao Differential Revision: D76273807 fbshipit-source-id: 88ac8c5c4d35288f0f4b1ec7f934d93cf10de666
1 parent e1ac345 commit c0670d4

File tree

8 files changed

+18
-2105
lines changed

8 files changed

+18
-2105
lines changed

.github/workflows/unittest_ci_cpu.yml

Lines changed: 18 additions & 27 deletions
Original file line numberDiff line numberDiff line change
@@ -20,33 +20,24 @@ on:
2020
jobs:
2121
build_test:
2222
strategy:
23-
fail-fast: false
24-
matrix:
25-
os:
26-
- linux.2xlarge
27-
python:
28-
- version: "3.9"
29-
tag: "py39"
30-
- version: "3.10"
31-
tag: "py310"
32-
- version: "3.11"
33-
tag: "py311"
34-
- version: "3.12"
35-
tag: "py312"
36-
- version: "3.13"
37-
tag: "py313"
38-
is_pr:
39-
- ${{ github.event_name == 'pull_request' }}
40-
exclude:
41-
- is_pr: true
42-
python:
43-
version: "3.10"
44-
- is_pr: true
45-
python:
46-
version: "3.11"
47-
- is_pr: true
48-
python:
49-
version: "3.12"
23+
fail-fast: false
24+
matrix:
25+
include:
26+
- os: linux.2xlarge
27+
python-version: 3.9
28+
python-tag: "py39"
29+
- os: linux.2xlarge
30+
python-version: '3.10'
31+
python-tag: "py310"
32+
- os: linux.2xlarge
33+
python-version: '3.11'
34+
python-tag: "py311"
35+
- os: linux.2xlarge
36+
python-version: '3.12'
37+
python-tag: "py312"
38+
- os: linux.2xlarge
39+
python-version: '3.13'
40+
python-tag: "py313"
5041
uses: pytorch/test-infra/.github/workflows/linux_job_v2.yml@main
5142
permissions:
5243
id-token: write

torchrec/distributed/hash_mc_embedding.py

Lines changed: 0 additions & 91 deletions
This file was deleted.

torchrec/distributed/tests/test_hash_zch_mc.py

Lines changed: 0 additions & 228 deletions
This file was deleted.

0 commit comments

Comments
 (0)