hpcaitech/FastFold

Trouble Running Perf.py in Benchmark

chuhan-ouyang opened this issue · 2 comments

chuhan-ouyang commented

Hi,

After the fix to the perf.py CLI issue, there is still a problem in running the file. It seems to be a mismatch between the size of the parameters between the layers.
image

Gy-Lu commented

Hi, but I can not reproduce this problem.
If you want to examine the shape, just print the tensor shape before this line:
image

But I think it might be something wrong with your environment.
Here is some discussion about this problem which you can refer:
https://discuss.pytorch.org/t/runtimeerror-cuda-error-cublas-status-invalid-value-when-calling-cublassgemm-handle-opa-opb-m-n-k-alpha-a-lda-b-ldb-beta-c-ldc/124544/18