From 18e2e9ae47e91c6bba47feac4407b9d2be77fd83 Mon Sep 17 00:00:00 2001 From: Yarong Mu Date: Fri, 9 May 2025 19:11:22 +0000 Subject: [PATCH] Add draft README as a starting point Signed-off-by: Yarong Mu --- .buildkite/pipeline.yml | 0 README.md | 27 +++++++++++++++++++++++++++ 2 files changed, 27 insertions(+) create mode 100644 .buildkite/pipeline.yml create mode 100644 README.md diff --git a/.buildkite/pipeline.yml b/.buildkite/pipeline.yml new file mode 100644 index 000000000..e69de29bb diff --git a/README.md b/README.md new file mode 100644 index 000000000..c8d26bcd7 --- /dev/null +++ b/README.md @@ -0,0 +1,27 @@ +Current structure, please add, modify, remove etc. + +tpu_commons/ +│── __init__.py +│── worker/ +│ ├── __init__.py +│ ├── tpu_worker.py # Moved and adapted from vllm/v1/worker/ +│ └── tpu_model_runner.py # Moved and adapted from vllm/v1/worker/ +│── attention/ +│ ├── __init__.py +│ └── backends/ +│ ├── __init__.py +│ └── pallas/ # <<< MOVED from vllm/v1/attention/backends/pallas/ +│ ├── __init__.py +│ ├── attention.py # (or whatever files are in original pallas dir) +│ └── metadata.py # (e.g., pallas_attention.py, pallas_metadata.py) +│── sample/ +│ ├── __init__.py +│ └── tpu/ # <<< MOVED from vllm/v1/sample/tpu/ +│ ├── __init__.py +│ ├── metadata.py +│ └── sampler.py +├── setup.py +├── pyproject.toml +└── .buildkite/ + └── pipeline.yml +