[ROCm] add support to ROCm 6.0 and MI300 (#2274)

2024-01-26 15:41:10 -05:00
parent 5265631d15
commit 6b7de1a030
8 changed files with 96 additions and 13 deletions
--- a/csrc/cuda_utils.h
+++ b/csrc/cuda_utils.h
@@ -5,3 +5,6 @@
 int get_device_attribute(
    int attribute,
    int device_id);
+
+int get_max_shared_memory_per_block_device_attribute(
+    int device_id);