misc/py-natten: New port: Neighborhood Attention Extension

misc/Makefile

@@ -454,6 +454,7 @@
     SUBDIR += py-mmsegmentation
     SUBDIR += py-molequeue
     SUBDIR += py-mxnet
+    SUBDIR += py-natten
     SUBDIR += py-npTDMS
     SUBDIR += py-numcodecs
     SUBDIR += py-onnx

misc/py-natten/Makefile

+PORTNAME=	natten
+DISTVERSION=	0.17.1
+CATEGORIES=	misc python
+MASTER_SITES=	PYPI
+PKGNAMEPREFIX=	${PYTHON_PKGNAMEPREFIX}
+
+MAINTAINER=	yuri@FreeBSD.org
+COMMENT=	Neighborhood Attention Extension
+WWW=		https://shi-labs.com/natten/
+
+LICENSE=	MIT
+LICENSE_FILE=	${WRKSRC}/LICENSE
+
+BUILD_DEPENDS=	cmake:devel/cmake-core \
+		pybind11>0:devel/pybind11 \
+		${PYTHON_PKGNAMEPREFIX}pytorch>=2.0.0:misc/py-pytorch@${PY_FLAVOR}
+LIB_DEPENDS=	libabsl_str_format_internal.so:devel/abseil \
+		libprotobuf.so:devel/protobuf
+RUN_DEPENDS=	${PYTHON_PKGNAMEPREFIX}packaging>=0:devel/py-packaging@${PY_FLAVOR} \
+		${PYTHON_PKGNAMEPREFIX}pytorch>=2.0.0:misc/py-pytorch@${PY_FLAVOR}
+
+USES=		python
+USE_PYTHON=	distutils autoplist
+
+post-install:
+	# strip binary
+	@${STRIP_CMD} ${STAGEDIR}${PYTHON_SITELIBDIR}/natten/libnatten${PYTHON_EXT_SUFFIX}.so
+	# delete stray files, see https://github.com/SHI-Labs/NATTEN/issues/152
+	@${RM} -r \
+		${STAGEDIR}${PYTHON_SITELIBDIR}/CMakeFiles \
+		${STAGEDIR}${PYTHON_SITELIBDIR}/CMakeCache.txt \
+		${STAGEDIR}${PYTHON_SITELIBDIR}/cmake_install.cmake \
+		${STAGEDIR}${PYTHON_SITELIBDIR}/Makefile
+
+.include <bsd.port.mk>

misc/py-natten/distinfo

+TIMESTAMP = 1721530370
+SHA256 (natten-0.17.1.tar.gz) = a2142722faa41fa3ae6d22e91ed527b2a6da5c4fe09a006f36d68edccaf9c643
+SIZE (natten-0.17.1.tar.gz) = 10848470

misc/py-natten/pkg-descr

+NATTEN is an open-source project dedicated to providing fast implementations
+for Neighborhood Attention, a sliding window self-attention mechanism.