NVIDIA / NVIDIA/cutlass

[QST] Where is exactly the definiton code of fprop convolution?

Open
#1,976 2 comments 0 reactions 0 assignees View on GitHub

Nobody has claimed this yet.

? - Needs Triage inactive-30d inactive-90d question
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

  1. Read the whole issue, then the project's contributing guide.
  2. Comment on the issue to say you are picking it up — it saves two people doing the same work.
  3. Fork the repository and make your change on a branch.
  4. 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

Get new issues in your inbox

A short digest of beginner-friendly GitHub issues.