Skip to content

GitLab

"src/array/cuda/csr_mm.hip" did not exist on "97fbd94dfc436ce8500770adced4a5f57889347c"
Switch branch/tag
Find file BlameHistoryPermalink
  • bmanga's avatar
    Ensure torchvision operators are added in C++ (#2798) · adfc15c4
    bmanga authored 
    * Ensure torchvision operators are registered in C++ via weak symbols

* Add note to README on how to ensure that torchvision operators are available in C++

* Fix dllimport/dllexport on windows, format files

* Factor out common macros in single file

* Expose cuda_version in the API, use it to avoid pruning of ops initializer
    adfc15c4
vision.cpp 2.37 KB