Add AV1 support

DALI_EXTRA_VERSION

@@ -1 +1 @@
-4d95e862cc8aa6495707a3a6d84cbf75dff812ef
+1ffbeaf1d085bb00f124038503508b3cb68e1a05

dali/test/python/decoder/test_video.py

@@ -33,6 +33,7 @@
 filenames = filter(lambda filename: "hevc" not in filename, filenames)
 # mpeg4 is not yet supported in the CPU operator itself
 filenames = filter(lambda filename: "mpeg4" not in filename, filenames)
+filenames = filter(lambda filename: "av1" not in filename, filenames)
 
 files = [np.fromfile(filename, dtype=np.uint8) for filename in filenames]
 
@@ -178,6 +179,7 @@ def input_gen(batch_size):
         filenames.append(f"{get_dali_extra_path()}/db/video/cfr_test.mp4")
         filenames = filter(lambda filename: "mpeg4" not in filename, filenames)
         filenames = filter(lambda filename: "hevc" not in filename, filenames)
+        filenames = filter(lambda filename: "av1" not in filename, filenames)
         filenames = cycle(filenames)
         while True:
             batch = []
@@ -211,6 +213,7 @@ def test_source_info(device):
     filenames = filter(lambda filename: "hevc" not in filename, filenames)
     # mpeg4 is not yet supported in the CPU operator itself
     filenames = filter(lambda filename: "mpeg4" not in filename, filenames)
+    filenames = filter(lambda filename: "av1" not in filename, filenames)
 
     files = list(filenames)
 

dali/test/python/input/test_video.py

@@ -27,6 +27,7 @@
 filenames = filter(lambda filename: "hevc" not in filename, filenames)
 # mpeg4 is not yet supported in the CPU operator
 filenames = filter(lambda filename: "mpeg4" not in filename, filenames)
+filenames = filter(lambda filename: "av1" not in filename, filenames)
 files = [np.fromfile(filename, dtype=np.uint8) for filename in filenames]
 
 batch_size_values = [1, 3, 100]

dali/test/python/test_dali_video_plugin_decoder.py

@@ -74,6 +74,19 @@ def test_cfr_h265_mp4_decoding():
     )
 
 
+def test_cfr_av1_mp4_decoding():
+    try:
+        run_video_decoding_test(
+            get_dali_extra_path() + "/db/video/cfr/test_1_av1.mp4",
+            f"{get_dali_extra_path()}/db/video/cfr/frames_1/frames_list.txt",
+            f"{get_dali_extra_path()}/db/video/cfr/frames_1",
+        )
+        return
+    except Exception as e:
+        assert "Codec not supported on this GPU" in str(e), "Unexpected error message: {}".format(e)
+        return
+
+
 def test_vfr_h264_mp4_decoding():
     run_video_decoding_test(
         get_dali_extra_path() + "/db/video/vfr/test_1.mp4",
@@ -88,3 +101,16 @@ def test_vfr_hevc_mp4_decoding():
         f"{get_dali_extra_path()}/db/video/vfr/frames_1_hevc/frames_list.txt",
         f"{get_dali_extra_path()}/db/video/vfr/frames_1_hevc",
     )
+
+
+def test_vfr_av1_mp4_decoding():
+    try:
+        run_video_decoding_test(
+            get_dali_extra_path() + "/db/video/vfr/test_1_av1.mp4",
+            f"{get_dali_extra_path()}/db/video/vfr/frames_1/frames_list.txt",
+            f"{get_dali_extra_path()}/db/video/vfr/frames_1",
+        )
+        return
+    except Exception as e:
+        assert "Codec not supported on this GPU" in str(e), "Unexpected error message: {}".format(e)
+        return