1 files changed, 56 insertions, 0 deletions
diff --git a/var/spack/repos/builtin/packages/rocpydecode/package.py b/var/spack/repos/builtin/packages/rocpydecode/package.py
new file mode 100644
index 0000000000..d4356625e2
--- /dev/null
+++ b/var/spack/repos/builtin/packages/rocpydecode/package.py
@@ -0,0 +1,56 @@
+# Copyright 2013-2024 Lawrence Livermore National Security, LLC and other
+# Spack Project Developers. See the top-level COPYRIGHT file for details.
+#
+# SPDX-License-Identifier: (Apache-2.0 OR MIT)
+
+from spack.package import *
+
+
+class Rocpydecode(CMakePackage):
+    """rocPyDecode is a set of Python bindings to rocDecode C++ library which provides
+    full HW acceleration for video decoding on AMD GPUs."""
+
+    homepage = "https://github.com/ROCm/rocPyDecode"
+    url = "https://github.com/ROCm/rocPyDecode/archive/refs/tags/rocm-6.2.0.tar.gz"
+
+    maintainers("afzpatel", "srekolam", "renjithravindrankannath")
+
+    version("6.2.4", sha256="9cdb8bdc65b54b2c02d6c950dd34cd702ec50d903aa4d252d1eb1f8cae8c0afb")
+    version("6.2.1", sha256="34c595cfe40ad74fcec2f52e7cc7be3ad8c8334030b0e98eb36305b6f63edc0d")
+    version("6.2.0", sha256="e465254cd3e96bbb59208e90293d7c6b7744b0fbcd928ef278ec568c83e63ff3")
+
+    depends_on("py-pybind11")
+    depends_on("ffmpeg@4.4:6")
+    depends_on("dlpack")
+
+    for ver in ["6.2.0", "6.2.1", "6.2.4"]:
+        depends_on(f"rocdecode@{ver}", when=f"@{ver}")
+
+    def patch(self):
+        filter_file(
+            r"${ROCM_PATH}/llvm/bin/clang++",
+            "{0}/bin/clang++".format(self.spec["llvm-amdgpu"].prefix),
+            "CMakeLists.txt",
+            string=True,
+        )
+        filter_file(
+            r"${ROCM_PATH}/share/rocdecode/utils",
+            "{0}/share/rocdecode/utils".format(self.spec["rocdecode"].prefix),
+            "CMakeLists.txt",
+            string=True,
+        )
+
+    def cmake_args(self):
+        args = [
+            self.define("rocDecode_PATH", self.spec["rocdecode"].prefix),
+            self.define("FFMPEG_INCLUDE_DIR", self.spec["ffmpeg"].prefix.include),
+            self.define("CMAKE_INSTALL_PREFIX_PYTHON", self.spec.prefix),
+            self.define("CMAKE_CXX_FLAGS", "-I{0}".format(self.spec["dlpack"].prefix.include)),
+            self.define(
+                "CMAKE_CXX_FLAGS",
+                "-DUSE_AVCODEC_GREATER_THAN_58_134 -I{0}".format(
+                    self.spec["dlpack"].prefix.include
+                ),
+            ),
+        ]
+        return args