How to set the corresponding maximum IO size based on different types of disks

HI：
In the development of HBA driver for Windows servers 2019 system, I need to set different maximum IO sizes for each disk. How to set the corresponding maximum IO size based on different types of disks. For example, SAS disk configuration is 512KB, NVMe disk configuration is 128KB