github/ROCm
1
1
Fork 0
mirror of https://github.com/ROCm/ROCm.git synced 2026-04-05 03:01:17 -04:00
Code Issues Packages Projects Releases Wiki Activity

[TUTORIALS] Add missing docstrings (#2420)

Browse Source 
Depend on https://github.com/openai/triton/pull/2419 to fix the
documentation workflow
This commit is contained in:
Keren Zhou
2023-09-29 17:24:30 -04:00
committed by GitHub
parent 90bef57acf
commit f2f5f1d457
1 changed files with 8 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

11
python/tutorials/11-grouped-gemm.py
View File

@@ -1,3 +1,11 @@
"""
Group GEMM
============================
This group gemm kernel launches a fixed number of CTA to compute a group
of gemms. The scheduling is static and we do it on device.
"""
# Copyright (c) 2023 NVIDIA Corporation & Affiliates. All rights reserved.
#
# Permission is hereby granted, free of charge, to any person obtaining
@@ -24,9 +32,6 @@ import torch
import triton
import triton.language as tl
# This group gemm kernel launches a fixed number of CTA to compute a group
# of gemms. The scheduling is static and we do it on device
@triton.autotune(
configs=[