vllm/csrc/cuda_utils.h at d06980dfa7e8dcc1738656beb46d3735c86faa21 - vllm - Gitea: Git with a cup of tea

wassname/vllm

mirror of https://github.com/wassname/vllm.git synced 2026-07-09 06:08:03 +08:00

Files

T

Yanming W e0c6f556e8 [Build] Avoid building too many extensions (#1624 )

2023-11-23 16:31:19 -08:00

6 lines

95 B

C

Raw Blame History

 #include <torch/extension.h>
 int get_device_attribute(
     int attribute,
     int device_id);