• Ashish Farmer's avatar
    [ROCm] Create torchvision as a HIP Extension (#1928) · 43e94b39
    Ashish Farmer authored
    * Added code to support creating extension on ROCm
    
    * max -> fmaxf conversion for hipification
    
    * added WITH_HIP flag for hipExtension
    
    * added appropriate headers for HIP build
    
    * use USE_ROCM in condition to build
    
    * change fmaxf and fminf calls
    
    * fminf -> min
    
    * fix the check for ROCM_HOME
    
    * more robust checking for rocm pytorch
    
    * add check for pytorch version before using HIP extensions
    
    * conditional reading of ROCM_HOME
    43e94b39
vision.cpp 1.38 KB