Tags: keyan/FBGEMM-1

Optimzed backward pass for ROCm devices (pytorch#3468)

Summary:
X-link: facebookresearch/FBGEMM#552


X-link: facebookresearch/FBGEMM#491

Added optimized implementation of backward pass for ROCm devices. Currently support **not nobag** mode, **rowwise_adagrad** optimizer with non-mixed dimensions in [64, 128, 160, 192].


Reviewed By: leitian

Differential Revision: D66310520

Pulled By: q10

wip

[fbgemm_gpu] Re-enable cache tests for ROCm

- Re-enable UVM cache tests for ROCm

Increase time-out for CUDA OSS CI (pytorch#3230)

Summary:
Pull Request resolved: pytorch#3230

X-link: facebookresearch/FBGEMM#328

As titled.

Due to Test time out for CUDA > 12 e.g., https://github.com/pytorch/FBGEMM/actions/runs/11223255193/job/31198763459

Reviewed By: q10

Differential Revision: D64005199

fbshipit-source-id: 339421d50b35665d0ccc2a909faef8720155cfdb

Increase time-out for CUDA OSS CI (pytorch#3230)

Summary:
Pull Request resolved: pytorch#3230

X-link: facebookresearch/FBGEMM#328

As titled.

Due to Test time out for CUDA > 12 e.g., https://github.com/pytorch/FBGEMM/actions/runs/11223255193/job/31198763459

Reviewed By: q10

Differential Revision: D64005199

fbshipit-source-id: 339421d50b35665d0ccc2a909faef8720155cfdb

Increase time-out for CUDA OSS CI (pytorch#3230)

Summary:
Pull Request resolved: pytorch#3230

X-link: facebookresearch/FBGEMM#328

As titled.

Due to Test time out for CUDA > 12 e.g., https://github.com/pytorch/FBGEMM/actions/runs/11223255193/job/31198763459

Reviewed By: q10

Differential Revision: D64005199

fbshipit-source-id: 339421d50b35665d0ccc2a909faef8720155cfdb

Add type error suppressions for upcoming upgrade (pytorch#3118)

Summary:
Pull Request resolved: pytorch#3118

X-link: facebookresearch/FBGEMM#206

Reviewed By: MaggieMoss

Differential Revision: D62550144

fbshipit-source-id: 792f706e61bc2cec8703ca4102afabbc26848a3b

Fix setuptools version (pytorch#2873)

Summary:
- Fix setuptools version to be >= 71.0.3 to get the fix for pypa/setuptools#4476 (affects Nova CI)

Pull Request resolved: pytorch#2873

Reviewed By: spcyppt

Differential Revision: D60074162

Pulled By: q10

fbshipit-source-id: c3f6589657b616e2366b10e4fc926bffc8569212

Fix setuptools version (pytorch#2873)

Summary:
- Fix setuptools version to be >= 71.0.3 to get the fix for pypa/setuptools#4476 (affects Nova CI)

Pull Request resolved: pytorch#2873

Reviewed By: spcyppt

Differential Revision: D60074162

Pulled By: q10

fbshipit-source-id: c3f6589657b616e2366b10e4fc926bffc8569212