aphrodite-engine/kernels/cuda_utils.h

#pragma once

int64_t get_device_attribute(int64_t attribute, int64_t device_id);

int64_t get_max_shared_memory_per_block_device_attribute(int64_t device_id);