search

torch / cutorch

A CUDA backend for Torch7
Other
336 stars 208 forks source link

The equal function for THCudaBlas_Sgemm in Aten #842

Open shen1994 opened 5 years ago

shen1994 commented 5 years ago

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 commented 5 years ago

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

Looottch commented 4 years ago

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 commented 4 years ago

ind the solution no

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

Looottch commented 4 years ago

ind the solution no

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

you use original code in the end?

shen1994 commented 4 years ago

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