[QST] Where is exactly the definiton code of fprop convolution?
Nobody has claimed this yet.
- Dominant language
- C++
- Stars
- 10.5k
- Forks
- 2.1k
- Avg merge
- 3d 11h
- Merged PRs (30d)
- 7
Description
What is your question?
I am studying example 16 and the hierarchy of fprop convolution within cutlass. I would like to know where is the low-level definition of the convolutions and their params.
Inside the example, the constructor used is linked with https://github.com/NVIDIA/cutlass/blob/main/include/cutlass/conv/device/implicit_gemm_convolution.h, but I cannot see there the source kernel or the structure of its parameters (activation, filters...). All I can see is a reference template constructor, but I don't know the code on which is based:
/// Argument structure
using Arguments = typename UnderlyingKernel::Arguments;
/// Kernel parameters object
typename UnderlyingKernel::Params params_;
I would like if it is used the constructor that the Convolution Readme links for the device (in my case GPU ampere), which is the following
Contributor guide
No contributing guide indexed for this repository
First steps
- Read the whole issue, then the project's contributing guide.
- Comment on the issue to say you are picking it up — it saves two people doing the same work.
- Fork the repository and make your change on a branch.
- Open a pull request that references the issue number.
Research direction
Start with examples/16_ampere_tensorop_conv2dfprop and include/cutlass/conv/device/implicit_gemm_convolution.h, then follow UnderlyingKernel::Arguments and Params into the implementation. Compare that path with media/docs/implicit_gemm_convolution.md and tools/util/include/cutlass/util/reference/device/convolution.h for the Ampere case. Done means identifying the low-level convolution definition and parameter structures requested.
Written by the indexing model from the issue text.
Assessment
- Tech stack
- cpp
- Domain
- machine-learning
- Issue type
- Documentation
- Difficulty
- 3/5
- Estimated time
- 1-2 days
- Activity status
- Stale
- Clarity
- Needs clarification
- Newbie friendliness
- 25/100