`nvdisasm`

class tests.tools.binaries.test_nvdisasm.TestFunctionView on GitHub 

REGISTERS: Final[dict[RegisterType, tuple[int, int]]] = {RegisterType.PRED: (1, 1), RegisterType.GPR: (8, 7), RegisterType.UGPR: (7, 3)}

test_string_representation() → NoneView on GitHub : Test string representation of reprospect.tools.binaries.nvdisasm.Function.

class tests.tools.binaries.test_nvdisasm.TestNVDisasmView on GitHub 

When there are many kernels.

CPP_FILE: Final[Path] = PosixPath('/__w/reprospect/reprospect/tests/tools/binaries/assets/many.cpp')

CUDA_FILE: Final[Path] = PosixPath('/__w/reprospect/reprospect/tests/tools/binaries/assets/many.cu')

SYMBOLS: Final[tuple[str, ...]] = ('_Z6say_hiv', '_Z20vector_atomic_add_42PKfS0_Pfj')

pytestmark = [Mark(name='parametrize', args=('parameters', (Parameters(arch=NVIDIAArch(family=<NVIDIAFamily.VOLTA: 'VOLTA'>, compute_capability=ComputeCapability(major=7, minor=0))), Parameters(arch=NVIDIAArch(family=<NVIDIAFamily.TURING: 'TURING'>, compute_capability=ComputeCapability(major=7, minor=5))), Parameters(arch=NVIDIAArch(family=<NVIDIAFamily.AMPERE: 'AMPERE'>, compute_capability=ComputeCapability(major=8, minor=0))), Parameters(arch=NVIDIAArch(family=<NVIDIAFamily.AMPERE: 'AMPERE'>, compute_capability=ComputeCapability(major=8, minor=6))), Parameters(arch=NVIDIAArch(family=<NVIDIAFamily.ADA: 'ADA'>, compute_capability=ComputeCapability(major=8, minor=9))), Parameters(arch=NVIDIAArch(family=<NVIDIAFamily.HOPPER: 'HOPPER'>, compute_capability=ComputeCapability(major=9, minor=0))), Parameters(arch=NVIDIAArch(family=<NVIDIAFamily.BLACKWELL: 'BLACKWELL'>, compute_capability=ComputeCapability(major=10, minor=0))), Parameters(arch=NVIDIAArch(family=<NVIDIAFamily.BLACKWELL: 'BLACKWELL'>, compute_capability=ComputeCapability(major=12, minor=0))))), kwargs={'ids': <class 'str'>})]

test_from_executable(workdir, parameters: Parameters, cmake_file_api: FileAPI) → NoneView on GitHub : Compile CPP_FILE as an executable, extract cubin and run nvdisasm.

class TestSaxpyView on GitHub 

When the kernel performs a saxpy.

CUDA_FILE: Final[Path] = PosixPath('/__w/reprospect/reprospect/tests/tools/assets/saxpy.cu')

SASS_ANNOTATED_FILE: Final[Path] = PosixPath('/__w/reprospect/reprospect/tests/tools/binaries/assets/saxpy.sass.annotated')

SIGNATURE: Final[str] = 'saxpy_kernel(float, const float *, float *, unsigned int)'

test_from_object(workdir, parameters: Parameters, cmake_file_api: FileAPI) → NoneView on GitHub : Compile CUDA_FILE as object, extract cubin and run nvdisasm.

test_from_sass_annotated() → NoneView on GitHub : Read annotated SASS from file and check reprospect.tools.binaries.NVDisasm.parse_sass_with_liveness_range_info().

test_string_representation() → NoneView on GitHub : Test reprospect.tools.binaries.NVDisasm.__str__().

nvdisasm