Skip to content
This repository was archived by the owner on Jun 3, 2025. It is now read-only.

[Remove] failing test #306

Merged
merged 1 commit into from
Apr 18, 2023
Merged

[Remove] failing test #306

Changes from all commits
Commits
Show all changes
1 commit
Select commit
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
21 changes: 1 addition & 20 deletions tests/sparsezoo/deployment_package/utils/test_extractors.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -17,10 +17,7 @@
import pytest

from sparsezoo import Model
from sparsezoo.deployment_package.utils.extractors import (
_batch_size_one_latency,
_throughput,
)
from sparsezoo.deployment_package.utils.extractors import _batch_size_one_latency


_VNNI_STUB = (
Expand Down Expand Up @@ -63,22 +60,6 @@ def model():
yield Model(source=stub)


@pytest.mark.parametrize(
"num_cores,batch_size,device_info,expected",
[
(24, 64, "c6i.12xlarge", 1948.45),
],
)
def test_throughput_extractor(model, num_cores, batch_size, device_info, expected):
actual_throughput = _throughput(
model=model,
num_cores=num_cores,
batch_size=batch_size,
device_info=device_info,
)
assert actual_throughput == expected


@patch("sparsezoo.deployment_package_module.utils.extractors._throughput")
def test_latency_extractor(throughput_patch, model):
throughput_patch.return_value = 0
Expand Down