OSSForks/pytorch
1
0
Fork 0
mirror of https://github.com/zebrajr/pytorch.git synced 2025-12-06 12:20:52 +01:00
Code Issues Actions 38 Packages Projects Releases Wiki Activity

fix bug in logging code (#150518)

Browse Source 
Fixes https://github.com/pytorch/pytorch/issues/150379

```python
>>> key = "aten._int_mm_1_2_3"
>>> m, n, k = key.split("_")[-3:]
>>> m, n, k
('1', '2', '3')
>>> name = "_".join(key.split("_")[:-3])
>>> name
'aten._int_mm'
```

Pull Request resolved: https://github.com/pytorch/pytorch/pull/150518
Approved by: https://github.com/xmfan
This commit is contained in:
Gabriel Ferns 2025-04-02 23:39:06 +00:00 committed by PyTorch MergeBot
parent 61a1f09b5b
commit 24f50653c8
1 changed files with 2 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
torch/_inductor/compile_fx.py
View File

@ -870,7 +870,8 @@ def _compile_fx_inner(
if log.isEnabledFor(logging.INFO):
mm_table_data = []
for key, value in counters["aten_mm_info"].items():
name, m, n, k = key.split("_")
m, n, k = key.split("_")[-3:]
name = "_".join(key.split("_")[:-3])
mm_table_data.append([name, m, n, k, value])
log.info("Overview info of inductor aten mms: ")
log.info(