-
|
Hi, I'm studying the first layer Convolution kernels mentioned here. And I find it is impossible for me to generate conv instances using the python interface on V100. So, just to be sure, are these first layer convolution kernels only available on arch >= SM80? |
Beta Was this translation helpful? Give feedback.
Replies: 1 comment 1 reply
-
|
sm75 and sm80 are tested, but i think sm70 should be supported out of the box. our |
Beta Was this translation helpful? Give feedback.
-
|
Thanks for your answer! That helps a lot. |
Beta Was this translation helpful? Give feedback.
sm75 and sm80 are tested, but i think sm70 should be supported out of the box.
our
generator.pydoes not have sm70, but you can use this one https://github.com/NVIDIA/cutlass/blob/master/tools/library/scripts/generator.py#L478 as a boilerplate and uses the tile sizes and instruction sizes in https://github.com/NVIDIA/cutlass/blob/master/tools/library/scripts/generator.py#L819 instead.