AI/rocm-systems
2
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

fixes bug with device id provided by user (#171)

Browse Source 
* fixes bug when setting env variable

* fix for device id

[ROCm/rocdecode commit: e7be5f2bbf]
This commit is contained in:
Lakshmi Kumar
2024-01-12 11:54:44 -08:00
committed by GitHub
parent d8c4040967
commit c7b2172448
2 changed files with 8 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
projects/rocdecode/samples/videoDecodePerf/videodecodeperf.cpp
+4 -1
View File
@@ -174,7 +174,10 @@ int main(int argc, char **argv) {
for (int i = 0; i < n_thread; i++) {
std::unique_ptr<VideoDemuxer> demuxer(new VideoDemuxer(input_file_path.c_str()));
rocDecVideoCodec rocdec_codec_id = AVCodec2RocDecVideoCodec(demuxer->GetCodecID());
v_device_id[i] = (i % 2 == 0) ? 0 : sd;
if (device_id % 2 == 0)
v_device_id[i] = (i % 2 == 0) ? device_id : device_id + sd;
else
v_device_id[i] = (i % 2 == 0) ? device_id - sd : device_id;
std::unique_ptr<RocVideoDecoder> dec(new RocVideoDecoder(v_device_id[i], mem_type, rocdec_codec_id, b_force_zero_latency, p_crop_rect));
v_demuxer.push_back(std::move(demuxer));
v_viddec.push_back(std::move(dec));