• q.yao's avatar
    Refactor csrc with device dispatcher (#1463) · 230f9a3b
    q.yao authored
    * Add device registry for pytorch ops
    
    * add declaration of CheckDeviceConsistency
    
    * fix for torch130
    
    * assert with torch check
    
    * Refactor ops with dispatch
    
    * update rest ops
    
    * faster install
    
    * update compatibility
    
    * update compatibility, rename parameter
    
    * move cpu implement to pytorch/cpu
    
    * update ops/csrc/README.md
    
    * fix rocm support
    
    * update cn document
    
    * update docs
    
    * list instead of map
    230f9a3b
setup.py 13.4 KB