The equal function for THCudaBlas_Sgemm in Aten

[shen1994]

What is the equal function for THCudaBlas_Sgemm in Aten??? May Someone tell me???Thanks~~~

THCudaBlas_Sgemm(state, ‘n’, ‘n’, n, m, k, 1.0f,
THCudaTensor_data(state, columns), n,
THCudaTensor_data(state, weight), k, 1.0f,
THCudaTensor_data(state, output_n), n);

[vandesa003]

Hi @shen1994 do you fix that problem? I'm facing the same problem. What is the THCudaTensor_data corresponding function in ATen?

[Looottch]

What is the equal function for THCudaBlas_Sgemm in Aten??? May Someone tell me???Thanks~~~

THCudaBlas_Sgemm(state, ‘n’, ‘n’, n, m, k, 1.0f,
THCudaTensor_data(state, columns), n,
THCudaTensor_data(state, weight), k, 1.0f,
THCudaTensor_data(state, output_n), n);

Have you find the solution now?

[shen1994]

ind the solution no

see https://github.com/shen1994/DepthAwareStereo

[Looottch]

ind the solution no

see https://github.com/shen1994/DepthAwareStereo

you use original code in the end?

[shen1994]

ind the solution no

see https://github.com/shen1994/DepthAwareStereo

you use original code in the end?

see testing ops, this will help you.
Cause cpp_extension has been used