24 lines
673 B
Python
24 lines
673 B
Python
# 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 PyThop(PythonPackage):
|
|
"""A tool to count the FLOPs of PyTorch model."""
|
|
|
|
homepage = "https://github.com/Lyken17/pytorch-OpCounter/"
|
|
url = "https://pypi.io/packages/py3/t/thop/thop-0.1.1.post2209072238-py3-none-any.whl"
|
|
|
|
license("MIT")
|
|
|
|
version(
|
|
"0.1.1.post2209072238",
|
|
sha256="01473c225231927d2ad718351f78ebf7cffe6af3bed464c4f1ba1ef0f7cdda27",
|
|
)
|
|
|
|
depends_on("py-torch", type=("build", "run"))
|