"src/vscode:/vscode.git/clone" did not exist on "7da8748cd8ffcee4bb437b32de3116af4db6904c"
  1. 23 May, 2025 2 commits
    • Yu Cheng's avatar
      [Dev] Add grouped GEMM backward example scripts (#515) · de028927
      Yu Cheng authored
      * Introduced `example_grouped_gemm_fwd.py` and `example_grouped_gemm_bwd.py` to demonstrate grouped matrix multiplication with forward and backward operations.
      * Implemented functions for grouped GEMM, input construction, and validation against PyTorch's implementation.
      * Added command-line argument parsing for flexible input configuration, including batch sizes and matrix dimensions.
      * Included a test function to validate the functionality with various input scenarios.
      de028927
    • Yu Cheng's avatar
      [Dev] Add grouped GEMM example with TileLang and PyTorch integration (#514) · fb801940
      Yu Cheng authored
      * Introduced a new example script `example_grouped_gemm.py` demonstrating grouped matrix multiplication using TileLang and PyTorch.
      * Implemented functions for performing grouped GEMM, constructing inputs, and validating results against PyTorch's implementation.
      * Added command-line argument parsing for flexible input configuration, including batch sizes and matrix dimensions.
      * Included a test function to validate the grouped GEMM functionality with various input scenarios.
      fb801940