[QST] How to pass a cute::Tensor as parameter to a device function?

[ZhangZhiPku]

I am writing a device function like:

__device__ void func(cute::Tensor tensor){...}

I want to call it like:

auto A = cute::make_tensor(...);
func(A);

However I got an error says: argument list for class template "cute::Tensor" is missing.
How can I pass a tensor as parameter to a function?

[ZhangZhiPku]

Besides, I rewrite this function to:

__device__ void func(float* tensor){...}

And call it like:

auto A = cute::make_tensor(...);
func(A.data());

There is another error says: no instance of function template "func" matches the argument list, argument types are cute::gmem_ptr<float *> rather than float .
How can I get the data pointer float from a tensor?

[thakkarV]

Tensor is a template type. The error you see "argument list for class template "cute::Tensor" is missing." is telling you what is wrong/missing. The signature of your device function will be something like this:

template <class Engine, class Layout>
auto
__device__
func(cute::Tensor<Engine, Layout> tensor) {
  // blah
}

[mnicely]

@ZhangZhiPku let us know if that works and we can close this issue

[ZhangZhiPku]

It works fine, thank you!