Initial project scaffold
This commit is contained in:
16
tasks/03_tiled_matmul/triton_skeleton.py
Normal file
16
tasks/03_tiled_matmul/triton_skeleton.py
Normal file
@@ -0,0 +1,16 @@
|
||||
"""Workbook-local Triton notes for tiled matmul."""
|
||||
|
||||
|
||||
def notes() -> str:
|
||||
return """
|
||||
TODO(student):
|
||||
1. Map one program instance to one output tile.
|
||||
2. Build row/col offsets for the tile.
|
||||
3. Loop over K in block_k chunks.
|
||||
4. Load A and B tiles, accumulate partial products.
|
||||
5. Store the output tile with masking on edges.
|
||||
"""
|
||||
|
||||
|
||||
if __name__ == "__main__":
|
||||
print(notes())
|
||||
Reference in New Issue
Block a user