Go binding for Pytorch C++ API (libtorch)
Geek Repo:Geek Repo
Github PK Tool:Github PK Tool
ZonghaoWang opened this issue 2 years ago · comments
for atc_cuda_device_count return int32 value, so the code should be change to
@ZonghaoWang ,
Thanks for the report!