Skip to content

Commit e3ce171

Browse files
authored
Mentioned cholesky in profiler docs (#75)
1 parent 16a858f commit e3ce171

2 files changed

Lines changed: 32 additions & 2 deletions

File tree

README.md

Lines changed: 9 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -7,7 +7,7 @@ Tested on linux and mac but should just work on Windows as well.
77

88
## New: Nsight Compute Profiling
99

10-
Profile QR v2 and `eigh` submissions on the hosted GPU Mode B200 Nsight Compute service.
10+
Profile submissions on the hosted GPU Mode B200 Nsight Compute service.
1111
See [docs/profiling.md](docs/profiling.md) for a complete copy-paste flow.
1212

1313
Quick QR v2 example:
@@ -26,6 +26,14 @@ export POPCORN_BREV_PROFILER_URL=https://http--brev-profiler-proxy--dxfjds728w5v
2626
popcorn submit submission.py --leaderboard eigh --profile-brev --benchmark-index 3 --no-tui
2727
```
2828

29+
Quick `cholesky` example:
30+
31+
```bash
32+
curl -O https://raw.githubusercontent.com/gpu-mode/reference-kernels/main/problems/linalg/cholesky_py/submission.py
33+
export POPCORN_BREV_PROFILER_URL=https://http--brev-profiler-proxy--dxfjds728w5v.code.run
34+
popcorn submit submission.py --leaderboard cholesky --profile-brev --benchmark-index 0 --no-tui
35+
```
36+
2937
The CLI downloads and extracts `ncu-details.txt` and `ncu-details.csv` for
3038
agent-readable analysis. It also extracts the optional `.ncu-rep` GUI report and
3139
ends with a macOS command that opens it in Nsight Compute:

docs/profiling.md

Lines changed: 23 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -95,7 +95,29 @@ Current `eigh` benchmark index table from `reference-kernels` main
9595
| 11 | `batch: 640; n: 512; cond: 0; seed: 780001; case: lapack_dense_even_spectrum` |
9696
| 12 | `batch: 60; n: 1024; cond: 0; seed: 780007; case: lapack_dense_geometric_spectrum` |
9797

98-
## 5. Read the Details
98+
## 5. Profile Cholesky
99+
100+
Get the `cholesky` starter submission:
101+
102+
```bash
103+
mkdir -p cholesky-profile
104+
cd cholesky-profile
105+
curl -O https://raw.githubusercontent.com/gpu-mode/reference-kernels/main/problems/linalg/cholesky_py/submission.py
106+
```
107+
108+
Profile the `batch=4096, n=32` benchmark:
109+
110+
```bash
111+
popcorn submit submission.py \
112+
--leaderboard cholesky \
113+
--profile-brev \
114+
--benchmark-index 0 \
115+
--no-tui
116+
```
117+
118+
The profiler will only profile the first 10 kernels, so if your solution uses more than that (the default Pytorch implementation does), then the profiler might be of limited use.
119+
120+
## 6. Read the Details
99121

100122
After the run finishes, the CLI downloads and extracts files like:
101123

0 commit comments

Comments
 (0)