Getting tvm-generated cuda codes

Questions
#1

Hi

I read tvm osdi paper and, for re-running experiments in the paper, found some method for automated search(they are in ‘autotvm’)

But i didn’t find the method that check where generated cuda cude(particularly, codes about convolution operation and gemm) is.

how can i find or get cuda codes generated by tvm?

If there is an answer, it will be helpful!

Thanks!

#2

You can have a look at an example of dumping generated cuda code here.

1 Like
#3

hi the link are not available， can u send the other link that show dumping generated cuda code please?

#4

Hi @lawpowen, this is likely what you’re looking for: Working with Operators Using Tensor Expression — tvm 0.9.dev0 documentation

#5

thank you so much , i will have a look to this doc!

#6

@carl-tercero

How to generate cuda code?

The code generated from the example is not cuda code.

For example, I can use tvm.build(ir, [A], target=‘c’) to generate c code.

But how to generate cuda code?

#7

Use target=‘cuda’

Here is an example

#8 
func = tvm.build(s, [A, B, C], target="cuda")
dev_module = func.imported_modules[0]
print(dev_module.get_source())