"docs/source/NAS/Benchmarks.rst" did not exist on "64c7f2983aba35d9cc7c2fee2210e86074a641e1"
  • Lei Wang's avatar
    [AMD][Docker] Create Dockerfile for ROCm environment setup (#355) · c58cbfbb
    Lei Wang authored
    * [Add] Create Dockerfile for ROCm environment setup
    
    * Introduced a new Dockerfile for setting up a ROCm environment with PyTorch.
    * Configured the working directory and installed necessary packages including Miniconda, Python, and development tools.
    * Cloned the tilelang repository and executed the ROCm installation script.
    * Set environment variables for compatibility and performance optimization.
    
    * [Remove] Delete Dockerfile for ROCm environment setup
    
    * Removed the Dockerfile used for setting up a ROCm environment with PyTorch.
    * Updated README to reflect changes in Docker image naming conventions for AMD GPU support.
    c58cbfbb
README.md 956 Bytes