"megatron/git@developer.sourcefind.cn:wuxk1/megatron-lm.git" did not exist on "630811b2c06fa7c7a5d51f3238c3ab7838f8eac7"
  • 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_cpu.h 3.1 KB