frozenleaves/pytorch
1
0
Fork 0
mirror of https://github.com/pytorch/pytorch.git synced 2025-10-21 13:44:15 +08:00
Code Issues Packages Projects Releases Wiki Activity

new pin (#97278)

Browse Source 
contains Fix for https://github.com/openai/triton/issues/1372

Pull Request resolved: https://github.com/pytorch/pytorch/pull/97278
Approved by: https://github.com/Skylion007
This commit is contained in:
Natalia Gimelshein
2023-03-28 23:30:23 +00:00
committed by PyTorch MergeBot
parent bc86af0d37
commit e0a647d8b5
2 changed files with 4 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
.ci/docker/ci_commit_pins/triton.txt
View File

@ -1 +1 @@
e650d3708be4dca12cc3491a2f8ab18ded47c368
46672772b46b103db7341c9e10fbad7f643557d4

6
torch/_inductor/kernel/mm_common.py
View File

@ -76,9 +76,9 @@ int8_mm_configs = functools.partial(
(32, 64, 32, 5, 8),
(128, 128, 32, 2, 8),
(64, 64, 64, 3, 8),
(32, 32, 128, 2, 4),
(64, 64, 16, 2, 4),
(32, 32, 16, 1, 2),
# (32, 32, 128, 2, 4),
# (64, 64, 16, 2, 4),
# (32, 32, 16, 1, 2),
(128, 256, 128, 3, 8),
(256, 128, 128, 3, 8),
),