• Hubert Lu's avatar
    Apex transformer (#77) · 27a47345
    Hubert Lu authored
    * Add setup_simple.py for debugging the compiling issue of scaled_masked_softmax_cuda
    
    * Comment out CUDA-specific implementations
    
    * Resolve filename collision of *cpp files with to-hipify code and *cu files
    27a47345
setup.py 33.5 KB