Optimize finding the compute partition mode (#493)

Этот коммит содержится в:
Aryan Salmanpour
2025-01-18 10:12:02 -05:00
коммит произвёл GitHub
родитель fe0fee01d0
Коммит 6ada77a9ad
2 изменённых файлов: 95 добавлений и 104 удалений
+2 -3
Просмотреть файл
@@ -153,7 +153,7 @@ private:
* retrieve the render node index for a given GPU UUID.
*/
std::unordered_map<std::string, int> gpu_uuids_to_render_nodes_map_;
std::unordered_map<std::string, ComputePartition> gpu_uuids_to_compute_partition_map_;
VaContext();
VaContext(const VaContext&) = delete;
VaContext& operator = (const VaContext) = delete;
@@ -162,7 +162,6 @@ private:
rocDecStatus InitHIP(int device_id, hipDeviceProp_t& hip_dev_prop);
rocDecStatus InitVAAPI(int va_ctx_idx, std::string drm_node);
void GetVisibleDevices(std::vector<int>& visible_devices_vetor);
void GetCurrentComputePartition(std::vector<ComputePartition> &current_compute_partitions);
void GetDrmNodeOffset(std::string device_name, uint8_t device_id, std::vector<int>& visible_devices, std::vector<ComputePartition> &current_compute_partitions, int &offset);
void GetDrmNodeOffset(std::string device_name, uint8_t device_id, std::vector<int>& visible_devices, ComputePartition current_compute_partition, int &offset);
void GetGpuUuids();
};