Code
Hub
Workspaces
Following
Trending
Connect
MCP
copy
Create free account
hub
/
github.com/NVIDIA/SOL-ExecBench
/ types & classes
Types & classes
95 in github.com/NVIDIA/SOL-ExecBench
⨍
Functions
597
◇
Types & classes
95
↳
Endpoints
5
↓ 28 callers
Class
ShiftingMemoryPoolAllocator
Pre-allocated memory pool that provides unique ``data_ptr`` per iteration. Allocates a buffer only slightly larger than the input tensors (ov
src/sol_execbench/core/bench/io.py:519
↓ 14 callers
Class
Workload
Concrete workload configuration for benchmarking. Defines a specific instance of a computational workload with concrete values for all variab
src/sol_execbench/core/data/workload.py:102
↓ 13 callers
Class
BenchmarkConfig
Configuration for benchmark runs. All fields have default values to make configuration optional.
src/sol_execbench/core/bench/config/benchmark_config.py:24
↓ 13 callers
Class
Trace
Complete trace linking a solution to a definition with evaluation results. A Trace represents the complete record of benchmarking a specific solu
src/sol_execbench/core/data/trace.py:176
↓ 11 callers
Class
Example
Parameters for one example e2e test case.
tests/examples/test_examples.py:51
↓ 10 callers
Class
Solution
A concrete implementation for a given Definition. Represents a complete solution that provides a high-performance implementation for a comput
src/sol_execbench/core/data/solution.py:234
↓ 8 callers
Class
Definition
Complete definition of a computational workload. A Definition provides a formal, machine-readable specification for a computational workload.
src/sol_execbench/core/data/definition.py:136
↓ 5 callers
Class
ClockPreset
GPU and DRAM clock frequencies for stable benchmarking.
src/sol_execbench/core/bench/config/device_config.py:25
↓ 5 callers
Class
Correctness
Correctness metrics from numerical evaluation. Contains error measurements comparing the solution output against a reference implementation t
src/sol_execbench/core/data/trace.py:27
↓ 5 callers
Class
ProblemPackager
Stage files for compilation and execution, returning commands for the CLI.
src/sol_execbench/driver/problem_packager.py:80
↓ 4 callers
Class
RewardHackDetected
Raised when a reward-hacking pattern is detected in a submission.
src/sol_execbench/core/bench/reward_hack.py:44
↓ 4 callers
Class
SupportedLanguages
Supported programming languages for solution implementations. Enumeration of programming languages that can be used to implement solutions fo
src/sol_execbench/core/data/solution.py:28
↓ 3 callers
Class
EvilCase
Parameters for one reward-hack e2e test case.
tests/sol_execbench/test_e2e.py:69
↓ 2 callers
Class
Sample
Parameters for one SOL ExecBench e2e test case.
tests/sol_execbench/test_e2e.py:58
↓ 2 callers
Class
ToleranceSpec
Tolerance specification for a workload. Specifies the conditions for a particular problem to be deemed numerically correct against the reference
src/sol_execbench/core/data/workload.py:81
↓ 1 callers
Class
BuildSpec
Build specification for a solution implementation. Contains all technical specifications required to build and execute a solution, including
src/sol_execbench/core/data/solution.py:131
↓ 1 callers
Class
Environment
Environment information from evaluation execution. Records the hardware and software environment details from when the evaluation was perform
src/sol_execbench/core/data/trace.py:73
↓ 1 callers
Class
Evaluation
Complete evaluation result for a solution on a workload. Records the full outcome of benchmarking a solution implementation against a specifi
src/sol_execbench/core/data/trace.py:113
↓ 1 callers
Class
Performance
Performance metrics from timing evaluation. Contains timing measurements and performance comparisons from benchmarking the solution against r
src/sol_execbench/core/data/trace.py:58
Class
AxisConst
Constant axis with a fixed value. A constant axis represents a dimension that has a fixed, compile-time known value. This is useful for dimen
src/sol_execbench/core/data/definition.py:35
Class
AxisExpr
Expression axis that can be specified at runtime. An expression axis represents a dimension whose value is determined at runtime based on a m
src/sol_execbench/core/data/definition.py:65
Class
AxisVar
Variable axis that can be specified at runtime. A variable axis represents a dimension whose value is determined at runtime based on the actu
src/sol_execbench/core/data/definition.py:51
Class
BaseModelWithDocstrings
Base model with the attribute docstrings being extracted to the model JSON schema.
src/sol_execbench/core/data/base_model.py:29
Class
CompileOptions
Compiler and linker flags for C++/CUDA solutions. Passed directly to the underlying build system (e.g. torch.utils.cpp_extension.load).
src/sol_execbench/core/data/solution.py:117
Class
CuptiKernelInfo
src/sol_execbench/core/bench/timing.py:101
Class
CustomInput
Custom input specification. Represents an that is generated from custom reference code. If a single input is this type, all inputs should be this
src/sol_execbench/core/data/workload.py:67
Class
DType
Supported data types for tensors. Enumeration of all data types that can be used in tensor specifications. Includes both floating-point and i
src/sol_execbench/core/data/definition.py:81
Class
EvaluationStatus
Status codes for evaluation results. Enumeration of all possible outcomes when evaluating a solution against a workload, covering success and
src/sol_execbench/core/data/trace.py:86
Class
MockEvent
tests/sol_execbench/core/bench/test_timing.py:97
Class
RandomInput
Random input generation descriptor. Represents a specification for generating random tensor input data during workload execution and benchmar
src/sol_execbench/core/data/workload.py:27
Class
SafetensorsInput
Input specification for data loaded from safetensors files. Represents tensor data that will be loaded from a safetensors file using a specif
src/sol_execbench/core/data/workload.py:51
Class
ScalarInput
Scalar literal input specification. Represents a scalar value (integer, float, or boolean) that will be used as a direct input parameter to t
src/sol_execbench/core/data/workload.py:38
Class
SourceFile
A single source code file in a solution implementation. Represents a source code file with its relative path and complete content. The file c
src/sol_execbench/core/data/solution.py:80
Class
SubTensor
tests/sol_execbench/core/bench/test_reward_hack.py:95
Class
SupportedBindings
Supported bindings for C++/CUDA solution implementations. Enumeration of binding types that can be used to interface compiled C++/CUDA code w
src/sol_execbench/core/data/solution.py:69
Class
SupportedHardware
Supported hardware targets for solution implementations. Enumeration of hardware platforms that solutions can target.
src/sol_execbench/core/data/solution.py:57
Class
TensorSpec
Specification for a tensor including shape and data type, to use as input or output of a kernel. This includes the symbolic shape (referencin
src/sol_execbench/core/data/definition.py:117
Class
TestAreClocksLocked
tests/sol_execbench/core/bench/test_clock_lock.py:360
Class
TestBenchGPUTimeWithCuptiGPU
GPU integration tests for CUPTI activity tracing timing accuracy. These tests mirror TestBenchTimeWithCUDAEventsGPU but use bench_gpu_time_wi
tests/sol_execbench/core/bench/test_timing.py:414
Class
TestBenchTimeWithCUDAEvents
tests/sol_execbench/core/bench/test_timing.py:92
Class
TestBenchTimeWithCUDAEventsGPU
GPU integration tests for do_bench timing accuracy. These tests verify that CUDA event timing correctly captures kernel execution time while
tests/sol_execbench/core/bench/test_timing.py:154
Class
TestCastToFP4x2
tests/sol_execbench/core/bench/test_io.py:99
Class
TestCheckLazyOutputs
tests/sol_execbench/core/bench/test_reward_hack.py:77
Class
TestCheckMonkeyPatch
tests/sol_execbench/core/bench/test_reward_hack.py:34
Class
TestCheckThreadInjection
tests/sol_execbench/core/bench/test_reward_hack.py:56
Class
TestClockLockValidation
tests/sol_execbench/core/bench/test_make_eval_clock_warn.py:67
Class
TestClockStatusMessage
tests/sol_execbench/core/bench/test_make_eval_clock_warn.py:43
Class
TestCloneArgs
tests/sol_execbench/core/bench/test_timing.py:42
Class
TestCompile
tests/sol_execbench/driver/test_problem_packager.py:186
Class
TestCompileOptions
Tests for compile option extraction.
tests/sol_execbench/driver/test_build_ext.py:177
Class
TestComputeErrorStats
Tests for compute_error_stats, focusing on near-zero edge cases.
tests/sol_execbench/core/bench/test_correctness.py:39
Class
TestConvertStdoutToTraces
tests/sol_execbench/driver/test_problem_packager.py:278
Class
TestCuptiVsCudaEvents
CUPTI should report lower or equal median times vs CUDA events. CUDA events include a fixed ~4-6us overhead from event recording. For short k
tests/sol_execbench/core/bench/test_timing.py:623
Class
TestDefinitionValidators
tests/sol_execbench/core/data/test_definition.py:89
Class
TestDtypeStrToPythonDtype
Tests for dtype_str_to_python_dtype.
tests/sol_execbench/core/data/test_dtypes.py:62
Class
TestDtypeStrToTorchDtype
Tests for dtype_str_to_torch_dtype (canonical source in data/dtypes.py).
tests/sol_execbench/core/data/test_dtypes.py:29
Class
TestEntryPointSuffixValidation
BuildSpec must reject entry points whose suffix doesn't match the language category.
tests/sol_execbench/core/data/test_solution.py:106
Class
TestEvalIntegrity
tests/sol_execbench/core/bench/test_reward_hack.py:112
Class
TestExecute
tests/sol_execbench/driver/test_problem_packager.py:241
Class
TestExtLoad
Tests for ext.load call arguments.
tests/sol_execbench/driver/test_build_ext.py:264
Class
TestGenInputs
tests/sol_execbench/core/bench/test_io.py:721
Class
TestGenerateHeuristicTensor
tests/sol_execbench/core/bench/test_io.py:251
Class
TestGetClockPreset
tests/sol_execbench/core/bench/test_clock_lock.py:377
Class
TestGetResolvedAxesValues
tests/sol_execbench/core/data/test_definition.py:42
Class
TestGetScalarInputs
tests/sol_execbench/core/data/test_workload.py:26
Class
TestIncludePaths
Tests for extra_include_paths construction.
tests/sol_execbench/driver/test_build_ext.py:220
Class
TestInit
tests/sol_execbench/driver/test_problem_packager.py:131
Class
TestIsBinaryMask
tests/sol_execbench/core/bench/test_io.py:184
Class
TestIsCausalAttentionMask
tests/sol_execbench/core/bench/test_io.py:165
Class
TestIsDtypeInteger
Tests for is_dtype_integer.
tests/sol_execbench/core/data/test_dtypes.py:94
Class
TestIsNormBias
tests/sol_execbench/core/bench/test_io.py:152
Class
TestIsNormWeight
tests/sol_execbench/core/bench/test_io.py:133
Class
TestIsPositiveTensor
tests/sol_execbench/core/bench/test_io.py:212
Class
TestIsRopeCosSin
tests/sol_execbench/core/bench/test_io.py:201
Class
TestIsSoftmaxOutput
tests/sol_execbench/core/bench/test_io.py:232
Class
TestIsSsmDecay
tests/sol_execbench/core/bench/test_io.py:223
Class
TestIsWeightMatrix
tests/sol_execbench/core/bench/test_io.py:117
Class
TestLanguageMixingValidation
BuildSpec must reject specs that mix C++ and Python languages.
tests/sol_execbench/core/data/test_solution.py:38
Class
TestLoadSafetensors
tests/sol_execbench/core/bench/test_io.py:801
Class
TestLockClocks
Tests for lock_clocks(). verify_clocks() and time.sleep() are mocked so we only test the locking logic. Verification is tested separately in
tests/sol_execbench/core/bench/test_clock_lock.py:78
Class
TestNormalizeOutputs
tests/sol_execbench/core/bench/test_io.py:657
Class
TestProbeClockLockAvailable
tests/sol_execbench/core/bench/test_clock_lock.py:41
Class
TestRandTensor
tests/sol_execbench/core/bench/test_io.py:48
Class
TestResolveBlobPath
tests/sol_execbench/core/bench/test_io.py:366
Class
TestSetSeed
Tests that set_seed produces reproducible GPU tensor generation.
tests/sol_execbench/core/bench/test_correctness.py:628
Class
TestShiftingMemoryPoolAllocator
tests/sol_execbench/core/bench/test_io.py:446
Class
TestSoRename
Tests for the .so rename logic after compilation.
tests/sol_execbench/driver/test_build_ext.py:241
Class
TestSourceDiscovery
Tests for source file collection logic.
tests/sol_execbench/driver/test_build_ext.py:142
Class
TestStreamHidingDetection
CUPTI-based timing must capture work hidden on non-default streams. A submission could launch kernels on a non-default stream to evade CUDA e
tests/sol_execbench/core/bench/test_timing.py:794
Class
TestSummarizeStatistics
tests/sol_execbench/core/bench/test_timing.py:75
Class
TestTemplateAST
Verify the raw template is valid Python.
tests/sol_execbench/driver/test_build_ext.py:119
Class
TestTimeRunnable
tests/sol_execbench/core/bench/test_timing.py:715
Class
TestUnlockClocks
tests/sol_execbench/core/bench/test_clock_lock.py:335
Class
TestVerifyClocks
tests/sol_execbench/core/bench/test_clock_lock.py:270
Class
_FakeTensor
tests/sol_execbench/samples/evil_lazy_output/kernel.py:25