MCPcopy Create free account

hub / github.com/NVIDIA/SOL-ExecBench / types & classes

Types & classes95 in github.com/NVIDIA/SOL-ExecBench

↓ 28 callersClassShiftingMemoryPoolAllocator
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 callersClassWorkload
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 callersClassBenchmarkConfig
Configuration for benchmark runs. All fields have default values to make configuration optional.
src/sol_execbench/core/bench/config/benchmark_config.py:24
↓ 13 callersClassTrace
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 callersClassExample
Parameters for one example e2e test case.
tests/examples/test_examples.py:51
↓ 10 callersClassSolution
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 callersClassDefinition
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 callersClassClockPreset
GPU and DRAM clock frequencies for stable benchmarking.
src/sol_execbench/core/bench/config/device_config.py:25
↓ 5 callersClassCorrectness
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 callersClassProblemPackager
Stage files for compilation and execution, returning commands for the CLI.
src/sol_execbench/driver/problem_packager.py:80
↓ 4 callersClassRewardHackDetected
Raised when a reward-hacking pattern is detected in a submission.
src/sol_execbench/core/bench/reward_hack.py:44
↓ 4 callersClassSupportedLanguages
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 callersClassEvilCase
Parameters for one reward-hack e2e test case.
tests/sol_execbench/test_e2e.py:69
↓ 2 callersClassSample
Parameters for one SOL ExecBench e2e test case.
tests/sol_execbench/test_e2e.py:58
↓ 2 callersClassToleranceSpec
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 callersClassBuildSpec
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 callersClassEnvironment
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 callersClassEvaluation
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 callersClassPerformance
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
ClassAxisConst
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
ClassAxisExpr
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
ClassAxisVar
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
ClassBaseModelWithDocstrings
Base model with the attribute docstrings being extracted to the model JSON schema.
src/sol_execbench/core/data/base_model.py:29
ClassCompileOptions
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
ClassCuptiKernelInfo
src/sol_execbench/core/bench/timing.py:101
ClassCustomInput
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
ClassDType
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
ClassEvaluationStatus
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
ClassMockEvent
tests/sol_execbench/core/bench/test_timing.py:97
ClassRandomInput
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
ClassSafetensorsInput
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
ClassScalarInput
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
ClassSourceFile
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
ClassSubTensor
tests/sol_execbench/core/bench/test_reward_hack.py:95
ClassSupportedBindings
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
ClassSupportedHardware
Supported hardware targets for solution implementations. Enumeration of hardware platforms that solutions can target.
src/sol_execbench/core/data/solution.py:57
ClassTensorSpec
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
ClassTestAreClocksLocked
tests/sol_execbench/core/bench/test_clock_lock.py:360
ClassTestBenchGPUTimeWithCuptiGPU
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
ClassTestBenchTimeWithCUDAEvents
tests/sol_execbench/core/bench/test_timing.py:92
ClassTestBenchTimeWithCUDAEventsGPU
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
ClassTestCastToFP4x2
tests/sol_execbench/core/bench/test_io.py:99
ClassTestCheckLazyOutputs
tests/sol_execbench/core/bench/test_reward_hack.py:77
ClassTestCheckMonkeyPatch
tests/sol_execbench/core/bench/test_reward_hack.py:34
ClassTestCheckThreadInjection
tests/sol_execbench/core/bench/test_reward_hack.py:56
ClassTestClockLockValidation
tests/sol_execbench/core/bench/test_make_eval_clock_warn.py:67
ClassTestClockStatusMessage
tests/sol_execbench/core/bench/test_make_eval_clock_warn.py:43
ClassTestCloneArgs
tests/sol_execbench/core/bench/test_timing.py:42
ClassTestCompile
tests/sol_execbench/driver/test_problem_packager.py:186
ClassTestCompileOptions
Tests for compile option extraction.
tests/sol_execbench/driver/test_build_ext.py:177
ClassTestComputeErrorStats
Tests for compute_error_stats, focusing on near-zero edge cases.
tests/sol_execbench/core/bench/test_correctness.py:39
ClassTestConvertStdoutToTraces
tests/sol_execbench/driver/test_problem_packager.py:278
ClassTestCuptiVsCudaEvents
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
ClassTestDefinitionValidators
tests/sol_execbench/core/data/test_definition.py:89
ClassTestDtypeStrToPythonDtype
Tests for dtype_str_to_python_dtype.
tests/sol_execbench/core/data/test_dtypes.py:62
ClassTestDtypeStrToTorchDtype
Tests for dtype_str_to_torch_dtype (canonical source in data/dtypes.py).
tests/sol_execbench/core/data/test_dtypes.py:29
ClassTestEntryPointSuffixValidation
BuildSpec must reject entry points whose suffix doesn't match the language category.
tests/sol_execbench/core/data/test_solution.py:106
ClassTestEvalIntegrity
tests/sol_execbench/core/bench/test_reward_hack.py:112
ClassTestExecute
tests/sol_execbench/driver/test_problem_packager.py:241
ClassTestExtLoad
Tests for ext.load call arguments.
tests/sol_execbench/driver/test_build_ext.py:264
ClassTestGenInputs
tests/sol_execbench/core/bench/test_io.py:721
ClassTestGenerateHeuristicTensor
tests/sol_execbench/core/bench/test_io.py:251
ClassTestGetClockPreset
tests/sol_execbench/core/bench/test_clock_lock.py:377
ClassTestGetResolvedAxesValues
tests/sol_execbench/core/data/test_definition.py:42
ClassTestGetScalarInputs
tests/sol_execbench/core/data/test_workload.py:26
ClassTestIncludePaths
Tests for extra_include_paths construction.
tests/sol_execbench/driver/test_build_ext.py:220
ClassTestInit
tests/sol_execbench/driver/test_problem_packager.py:131
ClassTestIsBinaryMask
tests/sol_execbench/core/bench/test_io.py:184
ClassTestIsCausalAttentionMask
tests/sol_execbench/core/bench/test_io.py:165
ClassTestIsDtypeInteger
Tests for is_dtype_integer.
tests/sol_execbench/core/data/test_dtypes.py:94
ClassTestIsNormBias
tests/sol_execbench/core/bench/test_io.py:152
ClassTestIsNormWeight
tests/sol_execbench/core/bench/test_io.py:133
ClassTestIsPositiveTensor
tests/sol_execbench/core/bench/test_io.py:212
ClassTestIsRopeCosSin
tests/sol_execbench/core/bench/test_io.py:201
ClassTestIsSoftmaxOutput
tests/sol_execbench/core/bench/test_io.py:232
ClassTestIsSsmDecay
tests/sol_execbench/core/bench/test_io.py:223
ClassTestIsWeightMatrix
tests/sol_execbench/core/bench/test_io.py:117
ClassTestLanguageMixingValidation
BuildSpec must reject specs that mix C++ and Python languages.
tests/sol_execbench/core/data/test_solution.py:38
ClassTestLoadSafetensors
tests/sol_execbench/core/bench/test_io.py:801
ClassTestLockClocks
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
ClassTestNormalizeOutputs
tests/sol_execbench/core/bench/test_io.py:657
ClassTestProbeClockLockAvailable
tests/sol_execbench/core/bench/test_clock_lock.py:41
ClassTestRandTensor
tests/sol_execbench/core/bench/test_io.py:48
ClassTestResolveBlobPath
tests/sol_execbench/core/bench/test_io.py:366
ClassTestSetSeed
Tests that set_seed produces reproducible GPU tensor generation.
tests/sol_execbench/core/bench/test_correctness.py:628
ClassTestShiftingMemoryPoolAllocator
tests/sol_execbench/core/bench/test_io.py:446
ClassTestSoRename
Tests for the .so rename logic after compilation.
tests/sol_execbench/driver/test_build_ext.py:241
ClassTestSourceDiscovery
Tests for source file collection logic.
tests/sol_execbench/driver/test_build_ext.py:142
ClassTestStreamHidingDetection
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
ClassTestSummarizeStatistics
tests/sol_execbench/core/bench/test_timing.py:75
ClassTestTemplateAST
Verify the raw template is valid Python.
tests/sol_execbench/driver/test_build_ext.py:119
ClassTestTimeRunnable
tests/sol_execbench/core/bench/test_timing.py:715
ClassTestUnlockClocks
tests/sol_execbench/core/bench/test_clock_lock.py:335
ClassTestVerifyClocks
tests/sol_execbench/core/bench/test_clock_lock.py:270
Class_FakeTensor
tests/sol_execbench/samples/evil_lazy_output/kernel.py:25