mirror of
https://github.com/vllm-project/vllm.git
synced 2025-10-20 23:03:52 +08:00
11 lines
184 B
C
11 lines
184 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);
|