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

PEP585 update - torch/nn torch/optim torch/package torch/profiler torch/serialization torch/sparse torch/xpu (#145175)

Browse Source 
See #145101 for details.
Pull Request resolved: https://github.com/pytorch/pytorch/pull/145175
Approved by: https://github.com/bobrenjc93
This commit is contained in:
Aaron Orenstein
2025-01-19 21:55:48 -08:00
committed by PyTorch MergeBot
parent bd97ce0b45
commit 54a00af2c6
38 changed files with 499 additions and 550 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
torch/sparse/_triton_ops.py
View File

@ -4,7 +4,7 @@ import math
import os
import weakref
from functools import lru_cache
from typing import Optional, Tuple
from typing import Optional
import torch
from torch._dynamo.utils import warn_once
@ -1124,7 +1124,7 @@ def _int_bsr_dense_addmm(
right_alpha: Optional[torch.Tensor] = None,
out: Optional[torch.Tensor] = None,
skip_checks: bool = False,
max_grid: Optional[Tuple[Optional[int], Optional[int], Optional[int]]] = None,
max_grid: Optional[tuple[Optional[int], Optional[int], Optional[int]]] = None,
meta: Optional[dict] = None,
):
if out is None and dense.dtype is torch.int8:
@ -1165,7 +1165,7 @@ def bsr_dense_addmm(
right_alpha: Optional[torch.Tensor] = None,
out: Optional[torch.Tensor] = None,
skip_checks: bool = False,
max_grid: Optional[Tuple[Optional[int], Optional[int], Optional[int]]] = None,
max_grid: Optional[tuple[Optional[int], Optional[int], Optional[int]]] = None,
meta: Optional[dict] = None,
):
"""Compute
@ -1647,7 +1647,7 @@ if has_triton():
alpha=1.0,
out: Optional[torch.Tensor] = None,
skip_checks: bool = False,
max_grid: Optional[Tuple[Optional[int], Optional[int], Optional[int]]] = None,
max_grid: Optional[tuple[Optional[int], Optional[int], Optional[int]]] = None,
):
f_name = "sampled_addmm"
@ -1731,7 +1731,7 @@ if has_triton():
*,
out: Optional[torch.Tensor] = None,
skip_checks: bool = False,
max_grid: Optional[Tuple[Optional[int], Optional[int], Optional[int]]] = None,
max_grid: Optional[tuple[Optional[int], Optional[int], Optional[int]]] = None,
meta: Optional[dict] = None,
):
f_name = "bsr_dense_mm"