Host targets device from config

Author: axiotisk

hls4ml/templates/vitis_accelerator/myproject_host.cpp

@@ -61,6 +61,24 @@ int main(int argc, char **argv) {
         return EXIT_FAILURE;
     }
 
+    std::string target_device = "myplatform";
+    auto device_type = target_device.substr(0, target_device.size() - 17);
+    std::cout << "Device type: " << device_type << std::endl;
+    std::cout << "xrt::device size " << sizeof(xrt::device) << std::endl;
+    for (int i = 0; i < sizeof(xrt::device); i++){
+      std::cout << "device[" << i << "] name:     " << xrt::device(i).get_info<xrt::info::device::name>() << "\n";
+      std::cout << "device[" << i << "] bdf:      " << xrt::device(i).get_info<xrt::info::device::bdf>() << "\n\n";
+      size_t found = xrt::device(i).get_info<xrt::info::device::name>().find(device_type);
+      if (found != std::string::npos){
+        std::cout << "Device: " << xrt::device(i).get_info<xrt::info::device::name>() << " found." << std::endl;
+        device_index = i;
+        std::cout << "Device index in loop: " <<device_index << std::endl;
+        break;
+      }
+        else{
+          std::cout << "Device not found" << std::endl;
+        }
+      }
     std::cout << "Open the device" << device_index << std::endl;
     auto device = xrt::device(device_index);
     std::cout << "Load the xclbin " << binaryFile << std::endl;

hls4ml/writer/vitis_accelerator_writer.py

@@ -113,6 +113,8 @@ def write_host(self, model):
         Args:
             model (ModelGraph): the hls4ml model.
         """
+        from hls4ml.backends import VitisAcceleratorConfig
+        vitis_accelerator_config = VitisAcceleratorConfig(model.config)
 
         filedir = os.path.dirname(os.path.abspath(__file__))
         f = open(os.path.join(filedir, '../templates/vitis_accelerator/myproject_host.cpp'))
@@ -127,6 +129,8 @@ def write_host(self, model):
                 newline = line.replace('myproject_kernel', format(model.config.get_project_name(), '_kernel'))
             elif 'output_dir' in line:
                 newline = line.replace('output_dir', format(model.config.get_output_dir()))
+            elif 'myplatform' in line:
+                newline = line.replace('myplatform', format(vitis_accelerator_config.get_platform()))
             else:
                 newline = line
             fout.write(newline)