MCPcopy Create free account
hub / github.com/NVIDIA/SOL-ExecBench / _make

Function _make

tests/sol_execbench/core/data/test_definition.py:26–36  ·  view source on GitHub ↗
(**overrides)

Source from the content-addressed store, hash-verified

24
25
26def _make(**overrides):
27 base = dict(
28 name="op",
29 op_type="gemm",
30 axes={"N": {"type": "var"}},
31 inputs={"a": {"shape": ["N"], "dtype": "float32"}},
32 outputs={"b": {"shape": ["N"], "dtype": "float32"}},
33 reference=_REFERENCE,
34 )
35 base.update(overrides)
36 return Definition(**base)
37
38
39# ── get_resolved_axes_values ──────────────────────────────────────────────────

Calls 1

DefinitionClass · 0.90

Tested by

no test coverage detected