You signed in with another tab or window. Reload to refresh your session.You signed out in another tab or window. Reload to refresh your session.You switched accounts on another tab or window. Reload to refresh your session.Dismiss alert
{{ message }}
Repository navigation
[Lane 2 · kotlin-core] TransposedConv2d — promote convTranspose2d to a @Diff op with backward rule #1374
Sub-issue of #1371 ([Feature]: TransposedConv2d — convTranspose2d op, CPU kernel, backward and StableHLO lowering).
Lane: 2 · Kotlin core Skill needed: Kotlin; some familiarity with the SKaiNET autodiff tape helps (it is small). No numerics research needed. Size: s (a few hours) Blocked by:#1373
What to do
Turn TensorOps.convTranspose2d into an abstract @Diff method like convTranspose1d (delete the throwing default). Fix every implementer the compiler complains about: VoidTensorOps (shape only — copy convTranspose1d and calculateConvTranspose1dShape), RecordingExecution, test fakes such as the one in ComputeGraphExecutorTest.
Add convTranspose2d to the conv-op bias list in skainet-lang/skainet-lang-ksp-processor/.../TracingWrapperProcessor.kt (search for convTranspose1d).
Add convTranspose2dBackward in skainet-compile/skainet-compile-dag/src/commonMain/kotlin/sk/ainet/lang/graph/DefaultExecutionTape.kt by copying convTranspose1dBackward, and register it in backwardDispatch (AutodiffCoverageTest fails until you do).
Run ./gradlew :skainet-compile:skainet-compile-dag:jvmTest :skainet-lang:skainet-lang-core:jvmTest apiDump, regenerate the matrix (./gradlew :skainet-compile:skainet-compile-hlo:jvmTest --tests '*HloCoverageDumpTest' :skainet-compile:skainet-compile-dag:jvmTest --tests '*BackwardCoverageDumpTest' generateDocs) and commit it.
Acceptance
Backward column shows ✅ for convTranspose2d
A gradient test compares against finite differences or PyTorch autograd
Result reported back on the parent issue
Notes
If something here turns out to be wrong or much bigger than the size label, comment on this issue or open a specific bug instead of blocking the parent.
Activity