Skip to content

Commit 2beb3ed

Browse files
author
Mark Saroufim
committed
Add missing submission.py for vectoradd_py problems
The popcorn-cli setup command expects every problem to have a submission.py, but pmpp/vectoradd_py and pmpp_v2/vectoradd_py only had solutions under correct/. Add simple PyTorch baselines so the CLI can bootstrap these problems.
1 parent a0dbafe commit 2beb3ed

2 files changed

Lines changed: 13 additions & 0 deletions

File tree

Lines changed: 6 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,6 @@
1+
from task import input_t, output_t
2+
3+
4+
def custom_kernel(data: input_t) -> output_t:
5+
A, B = data
6+
return A + B
Lines changed: 7 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,7 @@
1+
from task import input_t, output_t
2+
3+
4+
def custom_kernel(data: input_t) -> output_t:
5+
A, B, output = data
6+
output[...] = A + B
7+
return output

0 commit comments

Comments
 (0)