fix : use the passed in device-id

csrc/quantization/cutlass_w8a8/common.hpp

@@ -21,7 +21,7 @@ inline int get_cuda_max_shared_memory_per_block_opt_in(int const device) {
   int max_shared_mem_per_block_opt_in = 0;
   cudaDeviceGetAttribute(&max_shared_mem_per_block_opt_in,
                         cudaDevAttrMaxSharedMemoryPerBlockOptin,
-                        0);
+                        device);
   return max_shared_mem_per_block_opt_in;
 }