Remove unused broadcasting function

norabelrose · Apr 26, 2021 · a43532d · a43532d
1 parent 8c8c0db
commit a43532d
Showing 1 changed file with 0 additions and 8 deletions.
diff --git a/python/triton/ops/blocksparse/matmul.py b/python/triton/ops/blocksparse/matmul.py
@@ -711,11 +711,3 @@ def add_extra_dims(x):
  b = add_extra_dims(b)
 
  return a, b
-
-# Copied from the PyTorch 1.8 implementation so that we can support older PyTorch versions
-def broadcast_shapes(*shapes):
- with torch.no_grad():
- scalar = torch.zeros((), device="cpu")
- tensors = [scalar.expand(shape) for shape in shapes]
- tensors = torch.broadcast_tensors(*tensors)
- return tensors[0].shape