mirror of
https://github.com/wassname/vllm.git
synced 2026-06-27 16:17:48 +08:00
Michael Goin
8 lines
170 B
C
8 lines
170 B
C
#pragma once
|
|
|
|
#include <torch/extension.h>
|
|
|
|
int get_device_attribute(int attribute, int device_id);
|
|
|
|
int get_max_shared_memory_per_block_device_attribute(int device_id);
|