Skip to content

Commit

Permalink
update
Browse files Browse the repository at this point in the history
  • Loading branch information
akihironitta committed Jul 22, 2024
1 parent 85afc77 commit cc7a0cf
Showing 1 changed file with 1 addition and 2 deletions.
3 changes: 1 addition & 2 deletions pyg_lib/ops/__init__.py
Original file line number Diff line number Diff line change
Expand Up @@ -104,8 +104,7 @@ def grouped_matmul(
r"""Performs dense-dense matrix multiplication according to groups,
utilizing dedicated kernels that effectively parallelize over groups.
.. code-block:: python
Example:
inputs = [torch.randn(5, 16), torch.randn(3, 32)]
others = [torch.randn(16, 32), torch.randn(32, 64)]
Expand Down

0 comments on commit cc7a0cf

Please sign in to comment.