Remove all tritonbench code

Summary:
Tritonbench code is moved to pytorch-labs/tritonbench.
We can remove the tritonbench code from pytorch/benchmark.

Reviewed By: malfet, huydhn, kit1980

Differential Revision: D65548525

fbshipit-source-id: 625490a0fc4b060f64898357407a4e46643d3936

Author: xuzhao9

.ci/tritonbench/install-triton-nightly.sh

@@ -1,24 +1,10 @@
 name: TorchBench PR Test
 on:
   pull_request:
-    # ignore tritonbench paths
-    paths-ignore:
-      - 'torchbenchmark/operators/*'
-      - 'torchbenchmark/util/kernels/*'
-      - 'torchbenchmark/util/triton_op.py'
-      - 'userbenchmark/triton/*'
-      - '.ci/tritonbench/*'
   workflow_dispatch:
   push:
     branches:
       - main
-    # ignore tritonbench paths
-    paths-ignore:
-      - 'torchbenchmark/operators/*'
-      - 'torchbenchmark/util/kernels/*'
-      - 'torchbenchmark/util/triton_op.py'
-      - 'userbenchmark/triton/*'
-      - '.ci/tritonbench/*'
 
 jobs:
   cpu-test:

.ci/tritonbench/install.sh

@@ -4,21 +4,3 @@
 [submodule "submodules/lit-llama"]
 	path = submodules/lit-llama
 	url = https://github.com/Lightning-AI/lit-llama.git
-[submodule "submodules/FBGEMM"]
-	path = submodules/FBGEMM
-	url = https://github.com/pytorch/FBGEMM.git
-[submodule "submodules/cutlass-kernels"]
-	path = submodules/cutlass-kernels
-	url = https://github.com/ColfaxResearch/cutlass-kernels.git
-[submodule "submodules/flash-attention"]
-	path = submodules/flash-attention
-	url = https://github.com/Dao-AILab/flash-attention.git
-[submodule "submodules/kernels"]
-	path = submodules/kernels
-	url = https://github.com/triton-lang/kernels
-[submodule "submodules/ThunderKittens"]
-	path = submodules/ThunderKittens
-	url = https://github.com/HazyResearch/ThunderKittens.git
-[submodule "submodules/generative-recommenders"]
-	path = submodules/generative-recommenders
-	url = https://github.com/facebookresearch/generative-recommenders.git

.ci/tritonbench/test-install.sh

@@ -46,16 +46,6 @@ RUN cd /workspace/benchmark && \
     . ${SETUP_SCRIPT} && \
     python utils/cuda_utils.py --install-torchbench-deps
 
-# Install Tritonbench
-RUN cd /workspace/benchmark && \
-    bash .ci/tritonbench/install.sh
-
-# Test Tritonbench (libcuda.so.1 is required, so install libnvidia-compute-550 as a hack)
-RUN sudo apt update && sudo apt-get install -y libnvidia-compute-550 && \
-    cd /workspace/benchmark && \
-    bash .ci/tritonbench/test.sh && \
-    sudo apt-get purge -y libnvidia-compute-550
-
 # Install TorchAO benchmark
 RUN cd /workspace/benchmark && \
     . ${SETUP_SCRIPT} && \

.ci/tritonbench/test-operators.sh

@@ -76,7 +76,7 @@ def do_bench_kineto(
     ]
     if profile_opts is None:
         profile_opts = DEFAULT_PROFILE_OPTS
-    prefix = f"tritonbench_{fn._name}"
+    prefix = f"torchbench_{fn._name}"
     name = f"{prefix}_{datetime.now().strftime('%Y%m%d_%H%M%S')}_{''.join(random.choices(string.digits, k=10))}.json"
     with profiler.profile(
         schedule=profiler.schedule(wait=0, warmup=n_warmup, active=1, repeat=1),