optimum-benchmark

View on PyPIReverse Dependencies (0)

0.5.0 optimum_benchmark-0.5.0-py3-none-any.whl

Wheel Details

Project: optimum-benchmark
Version: 0.5.0
Filename: optimum_benchmark-0.5.0-py3-none-any.whl
Download: [link]
Size: 105897
MD5: d94297fbcf57b26e1d9938fae9d64f1e
SHA256: 3db791c635b3873e38459e777c295d99db4e6179e6d4dc4bbcdcf01f127429a3
Uploaded: 2025-01-24 17:48:04 +0000

dist-info

METADATA

Metadata-Version: 2.1
Name: optimum-benchmark
Version: 0.5.0
Summary: Optimum-Benchmark is a unified multi-backend utility for benchmarking Transformers, Timm, Diffusers and Sentence-Transformers with full support of Optimum's hardware optimizations & quantization schemes.
Author: HuggingFace Inc. Special Ops Team
Home-Page: https://github.com/huggingface/optimum-benchmark
License: Apache
Keywords: benchmark,transformers,quantization,pruning,optimization,training,inference,onnx,onnx runtime,intel,habana,graphcore,neural compressor,ipex,ipu,hpu,llm-swarm,py-txi,vllm,llama-cpp,auto-gptq,autoawq,sentence-transformers,bitsandbytes,codecarbon,flash-attn,deepspeed,diffusers,timm,peft
Classifier: Intended Audience :: Education
Classifier: Intended Audience :: Developers
Classifier: Operating System :: POSIX :: Linux
Classifier: Intended Audience :: Science/Research
Classifier: Programming Language :: Python :: 3.8
Classifier: Programming Language :: Python :: 3.9
Classifier: Programming Language :: Python :: 3.10
Classifier: License :: OSI Approved :: Apache Software License
Classifier: Topic :: Scientific/Engineering :: Artificial Intelligence
Requires-Dist: transformers
Requires-Dist: accelerate
Requires-Dist: datasets
Requires-Dist: hydra-core
Requires-Dist: omegaconf
Requires-Dist: psutil
Requires-Dist: typing-extensions
Requires-Dist: flatten-dict
Requires-Dist: colorlog
Requires-Dist: pandas
Requires-Dist: rich
Requires-Dist: nvidia-ml-py
Requires-Dist: optimum; extra == "auto-gptq"
Requires-Dist: auto-gptq; extra == "auto-gptq"
Requires-Dist: autoawq; extra == "autoawq"
Requires-Dist: bitsandbytes; extra == "bitsandbytes"
Requires-Dist: codecarbon; extra == "codecarbon"
Requires-Dist: deepspeed; extra == "deepspeed"
Requires-Dist: diffusers; extra == "diffusers"
Requires-Dist: flash-attn; extra == "flash-attn"
Requires-Dist: optimum[ipex] (>=1.18.0); extra == "ipex"
Requires-Dist: llama-cpp-python; extra == "llama-cpp"
Requires-Dist: llm-swarm; extra == "llm-swarm"
Requires-Dist: optimum[onnxruntime] (>=1.18.0); extra == "onnxruntime"
Requires-Dist: optimum[onnxruntime-gpu] (>=1.18.0); extra == "onnxruntime-gpu"
Requires-Dist: optimum[nncf,openvino] (>=1.18.0); extra == "openvino"
Requires-Dist: peft; extra == "peft"
Requires-Dist: py-txi; extra == "py-txi"
Requires-Dist: ruff; extra == "quality"
Requires-Dist: sentence-transformers; extra == "sentence-transformers"
Requires-Dist: optimum[nvidia] (>=1.18.0); extra == "tensorrt-llm"
Requires-Dist: pytest; extra == "testing"
Requires-Dist: hydra-joblib-launcher; extra == "testing"
Requires-Dist: timm; extra == "timm"
Requires-Dist: torch-ort; extra == "torch-ort"
Requires-Dist: onnxruntime-training; extra == "torch-ort"
Requires-Dist: optimum (>=1.18.0); extra == "torch-ort"
Requires-Dist: torchao; extra == "torchao"
Requires-Dist: vllm; extra == "vllm"
Provides-Extra: auto-gptq
Provides-Extra: autoawq
Provides-Extra: bitsandbytes
Provides-Extra: codecarbon
Provides-Extra: deepspeed
Provides-Extra: diffusers
Provides-Extra: flash-attn
Provides-Extra: ipex
Provides-Extra: llama-cpp
Provides-Extra: llm-swarm
Provides-Extra: onnxruntime
Provides-Extra: onnxruntime-gpu
Provides-Extra: openvino
Provides-Extra: peft
Provides-Extra: py-txi
Provides-Extra: quality
Provides-Extra: sentence-transformers
Provides-Extra: tensorrt-llm
Provides-Extra: testing
Provides-Extra: timm
Provides-Extra: torch-ort
Provides-Extra: torchao
Provides-Extra: vllm
Description-Content-Type: text/markdown
License-File: LICENSE
[Description omitted; length: 19953 characters]

WHEEL

Wheel-Version: 1.0
Generator: bdist_wheel (0.41.2)
Root-Is-Purelib: true
Tag: py3-none-any

RECORD

Path Digest Size
optimum_benchmark/__init__.py sha256=A6wfeezfwlCJh_LVT1ShpCB4aZczW-h4TjACYD_EWx8 950
optimum_benchmark/cli.py sha256=gcGzFG_Lx4S45lxPg4nX3P5YNsQX8ma6nlAf3D8Shdo 3053
optimum_benchmark/hub_utils.py sha256=R93QgergEBBuaBSi-G2ySksVLtWktxBLPDO559RpLCc 5387
optimum_benchmark/import_utils.py sha256=oZAOIdiojYwf_9SNJsSRemGQB5ff9F5EfI1i_RcoMEc 7925
optimum_benchmark/logging_utils.py sha256=iP4B4gZW7L3ILTWIuUPXgTrw4-AYOIkHvACzYPJaLYM 2222
optimum_benchmark/process_utils.py sha256=gtNdFV1y6KV5AUG_HR-GNOXxI3Pp9TLDmLOSPiqntDQ 290
optimum_benchmark/system_utils.py sha256=ZSuihGTOSzKb_fEZZ8mAmlpQPoopQ1sW8uSuqjUGD34 8361
optimum_benchmark/task_utils.py sha256=wSM7Qj1UOY34mEG9ZcS1hnWiJWJ8B47F5Q-XCoIJoXU 12283
optimum_benchmark/version.py sha256=R-76FqRp_T5t1h3QednHC1Qsn4Jdgnuh_BhmfHuVZ1U 639
optimum_benchmark/backends/__init__.py sha256=3JJzbYpM_y5c7EaND6BurX_QZgskomewNa612wTted4 612
optimum_benchmark/backends/base.py sha256=6RI1L1xYVeSqId9liLG2ECdRMY1g3OeOW2e6Tpg5YmQ 6307
optimum_benchmark/backends/config.py sha256=JrKH_uu1rNEJRdS5UY07PxUNi3CZsRY5Zh-gQFuLW3k 5268
optimum_benchmark/backends/diffusers_utils.py sha256=7P8i2BB5NRDRQ78OQh2rfQu_Tl1_uF0en-uucx-3JdQ 1317
optimum_benchmark/backends/peft_utils.py sha256=zclPExVGf626tZEwkauI8thYHHuQt26f-KAS5uXhJ1E 581
optimum_benchmark/backends/timm_utils.py sha256=Quu79Ki0WHX0Zgs_Fv84DFHT3Fwo2HcIXKIsBPoFqC0 2671
optimum_benchmark/backends/transformers_utils.py sha256=9-QpDUQ6yGYLvz9-PxwkMJGrs8GUFmllXCaSTeC5HTM 8718
optimum_benchmark/backends/ipex/__init__.py sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU 0
optimum_benchmark/backends/ipex/backend.py sha256=H2TwnzAUhGtPTsI-kjQi5rMm8rRut-H2B926gnVDmkg 3565
optimum_benchmark/backends/ipex/config.py sha256=670KmQacl0029GAup1a78OhVCR3Dk3HQHXCcI39hCYg 1248
optimum_benchmark/backends/ipex/utils.py sha256=E9QYmZHrimKJvCwfm0Z5qLzaiPR4cp20xMaDyFywQQ8 578
optimum_benchmark/backends/llama_cpp/__init__.py sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU 0
optimum_benchmark/backends/llama_cpp/backend.py sha256=BMLEGafynwwTj92MiF8OXDURg18GRM2Zyh19pjmNxbs 2390
optimum_benchmark/backends/llama_cpp/config.py sha256=Elg4tQokq3Lb6_GclGV4jF4BCfTPXF1TqRxMLSwWnOY 883
optimum_benchmark/backends/onnxruntime/__init__.py sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU 0
optimum_benchmark/backends/onnxruntime/backend.py sha256=11lhwdxrtTYMsdyebcrDU8whuWY4hkSkRs7ot0VeYfQ 13706
optimum_benchmark/backends/onnxruntime/config.py sha256=aF9KX7jQvVVRcj91DUrzm2CzSDpLL3DALxnQQ-70oig 4198
optimum_benchmark/backends/onnxruntime/utils.py sha256=CLu57RJHwz8D4PweCeCwfC_o0qMIoR3atxVGMXNooBs 1747
optimum_benchmark/backends/openvino/__init__.py sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU 0
optimum_benchmark/backends/openvino/backend.py sha256=ReGWLtpJzJ2YGAVn15ILcJu8lLhDzCfbZnaU3Zs8d_4 4957
optimum_benchmark/backends/openvino/config.py sha256=SlkgjZfJq6GZcnOqbknb6JjPNXSBMAza257Zai-7rOA 1395
optimum_benchmark/backends/openvino/utils.py sha256=9XkGoptDrS28qpt83RQDZSkdedDpQ5I3SKLn32wompg 1030
optimum_benchmark/backends/py_txi/__init__.py sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU 0
optimum_benchmark/backends/py_txi/backend.py sha256=LuwUHE5yZhOOQF__mJLbJxrKfVEgsUdv6dDSIXZxn5I 7422
optimum_benchmark/backends/py_txi/config.py sha256=1-_CljDJO6QfzqHf7jZsZ0Dij9htl0FaOacDD0bl5Iw 2503
optimum_benchmark/backends/pytorch/__init__.py sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU 0
optimum_benchmark/backends/pytorch/backend.py sha256=ClbVsUc9BHOgR_tRFdnmE6qx0JiSu-7jhny9ebiJxoI 21745
optimum_benchmark/backends/pytorch/config.py sha256=SyWbr3e_1ivmTJT4k1FA9Ws34yjVX_95B-YYKBjqoaY 3364
optimum_benchmark/backends/tensorrt_llm/__init__.py sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU 0
optimum_benchmark/backends/tensorrt_llm/backend.py sha256=jHhHHqgpzodL934wquDpFs5o1x2JXMoMAsO9FJ7DpnA 7347
optimum_benchmark/backends/tensorrt_llm/config.py sha256=nxwjh2cM-cLJSTSOS7XlkwnQqdBnQm92i3a8IFOc8Cw 1837
optimum_benchmark/backends/tensorrt_llm/utils.py sha256=ToFWs74l2UlP8NGmiHsoWwe-fcYEHd7wJFgFctNfl2s 87
optimum_benchmark/backends/torch_ort/__init__.py sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU 0
optimum_benchmark/backends/torch_ort/backend.py sha256=-E9GCd2f4nfjPE_fv5DPiHzgJ1Znw1E2AN1c78Nyfwc 3220
optimum_benchmark/backends/torch_ort/config.py sha256=JNvoDp3_zqISTck7BegEOOB90FlA8wAkjxWdT6SkIJ0 928
optimum_benchmark/backends/vllm/__init__.py sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU 0
optimum_benchmark/backends/vllm/backend.py sha256=ezSsUqrpgfTJz8WdNNcS-Zj9dEgbuqHJYcOm6_I7-G4 8133
optimum_benchmark/backends/vllm/config.py sha256=WLIO12KZcd8N6g7LhqN7KUE9KnhqdUx_Bzds_0cHuKY 2216
optimum_benchmark/benchmark/__init__.py sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU 0
optimum_benchmark/benchmark/base.py sha256=0yRsKG9k2mkPbj9_5De78x1cv4YmGAoQkR8uiGLiQ40 2686
optimum_benchmark/benchmark/config.py sha256=opiFEtYls2I2mKgKCfRApqsDhLjAcKlCGsKtlOpXqcY 957
optimum_benchmark/benchmark/report.py sha256=6iExoIQ8PwwpKZOXuW3akCeq_fAyISj_z26RQ95z67I 5618
optimum_benchmark/generators/__init__.py sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU 0
optimum_benchmark/generators/base.py sha256=r3OBHW_fXy5IPH0tSZz2VRZnjIRr58fpLLk8L2BmmJ4 1895
optimum_benchmark/generators/dataset_generator.py sha256=a_xYm9D22Kg_7Im1Qdp12IOaXCrh2nkxEaqeOW7bT80 1618
optimum_benchmark/generators/input_generator.py sha256=slfJsILfCAy8T_tHu0aDSYPBoTMBKUi_OD7HWuXT1A4 1371
optimum_benchmark/generators/model_generator.py sha256=fRu6CqeKXZTqYrXH5O-yuK8cuwTlBYnvwIjQRSoehzM 8404
optimum_benchmark/generators/task_generator.py sha256=5KxaTYPdm_vDLEtFGV1AKDlKnJukm50gPS-y51-U3Zo 13557
optimum_benchmark/launchers/__init__.py sha256=kixBt65vbRGzawMM0CSxELRoMDzhv00sRR5okcTFqTE 317
optimum_benchmark/launchers/base.py sha256=rwL_y70SBPW8fH2Z-ZV188XX5LgvWRPpSefU0kxpGnU 3717
optimum_benchmark/launchers/config.py sha256=fLS4lnvFTyWHumQY7IEDwzbrG4oAgTA6_Zjs-jKS3R0 1663
optimum_benchmark/launchers/device_isolation_utils.py sha256=r8fGbx5hbf0D8IAWHsCtNOHs168-hVUqYCaF5SH2qsY 6352
optimum_benchmark/launchers/inline/__init__.py sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU 0
optimum_benchmark/launchers/inline/config.py sha256=zsjexgb1cm13ZmdapVFa_jYZDQHTkxtO5RSvs4N9PnY 685
optimum_benchmark/launchers/inline/launcher.py sha256=j51ixjGrV7Rcy8e0JCTHSTUtsCikVJour21q4TNxGWI 584
optimum_benchmark/launchers/process/__init__.py sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU 0
optimum_benchmark/launchers/process/config.py sha256=WyL3LZDIFCoaqFihV5swue7m7bkhqN2S8Kwe26mxL-E 485
optimum_benchmark/launchers/process/launcher.py sha256=Pu9jejqcc7_EjKV1We6Qh6pODB7HSvCNYJjnGnjLjfU 4149
optimum_benchmark/launchers/torchrun/__init__.py sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU 0
optimum_benchmark/launchers/torchrun/config.py sha256=lMiB-yFTTP8pIes-hT8q09_3_Xw9roxLRj0J-72tlsE 2404
optimum_benchmark/launchers/torchrun/launcher.py sha256=xKEm2sslO08kJOT5VK5cw0ed_a6IuTMFO54y1yC0CbU 6806
optimum_benchmark/profilers/__init__.py sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU 0
optimum_benchmark/profilers/fx_profiler.py sha256=ughThut-_zzwnRSYH-jzi8tmLuA1MuhWcQVAeKYsKHI 1538
optimum_benchmark/profilers/ort_profiler.py sha256=XojvyslGvZCuZkK6uOq7GhVsfH3J0_DaVrNAzhNpiqU 1958
optimum_benchmark/scenarios/__init__.py sha256=XyT2jqCDiV1OwAhAvqiS9jiLmJWhHMXgpGmsJdD5asI 336
optimum_benchmark/scenarios/base.py sha256=4T27_XmG2rYYCuUUrywm7hI-uWb8_0Pth3CcTNnGa3g 600
optimum_benchmark/scenarios/config.py sha256=YRDaFATJZK0SAxNRAJSNh2kMoB0FVNzDAVJrNih0xfo 328
optimum_benchmark/scenarios/energy_star/__init__.py sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU 0
optimum_benchmark/scenarios/energy_star/config.py sha256=R8dM3x-3qMmNsAy6eIcxyC1RkFOtWtAIScXqpSfLllQ 5130
optimum_benchmark/scenarios/energy_star/scenario.py sha256=qpAk-FJLASkGMLiXDGgi5UTY2D0QxS3TawA11OXYO5o 15335
optimum_benchmark/scenarios/inference/__init__.py sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU 0
optimum_benchmark/scenarios/inference/config.py sha256=goQt7B0ZV_CJTXpI4hVVLRHGk50AsWE8QfjLnv0tefs 4363
optimum_benchmark/scenarios/inference/scenario.py sha256=g8u7US8MmBCkaNfs8E01fiHpr69w7wese4SdmfKlvdM 19131
optimum_benchmark/scenarios/training/__init__.py sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU 0
optimum_benchmark/scenarios/training/config.py sha256=nbi0AnDBk0wSXvRzmFMJVxPN1H5-Hfbkgo1z96CnsMs 2625
optimum_benchmark/scenarios/training/scenario.py sha256=fRth0pzWk-vgWWTErmbNgzl1tT8W_iTbHRbedmKuLK8 4934
optimum_benchmark/trackers/__init__.py sha256=edT3AN7mY-2GIyEFW6lGcZbkjyc9j9nCCy_2iucE8aY 668
optimum_benchmark/trackers/energy.py sha256=41j7MTpDVkmrGJIe2s2Q2UaCnf6JOmTw25goIEevcvo 10167
optimum_benchmark/trackers/latency.py sha256=5Lux6Cv9DyWL1dINcqILjHpIEi_i59wnoGEnOCCMyRk 22462
optimum_benchmark/trackers/memory.py sha256=JCzzYIVlppo1PxvzNppQsbJ6uGv8QbfPyfkt1vK4-CM 16360
optimum_benchmark-0.5.0.dist-info/LICENSE sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ 11357
optimum_benchmark-0.5.0.dist-info/METADATA sha256=jvBGXmYg4ia1n85SMd_JI-OX8ai8idpMLIF9nxoUhM0 23584
optimum_benchmark-0.5.0.dist-info/WHEEL sha256=yQN5g4mg4AybRjkgi-9yy4iQEFibGQmlz78Pik5Or-A 92
optimum_benchmark-0.5.0.dist-info/entry_points.txt sha256=KGPzkl9fQ2VDWkPZWUNJGZtMnlLMINRFJmEq5Iezxj0 65
optimum_benchmark-0.5.0.dist-info/top_level.txt sha256=ux6J-4WQAFRha1EphTMW2k2qlOk5iOsdsGqmE0JyN_8 18
optimum_benchmark-0.5.0.dist-info/RECORD

top_level.txt

optimum_benchmark

entry_points.txt

optimum-benchmark = optimum_benchmark.cli:main