python

Func Class — pytorch Architecture

Architecture documentation for the Func class in bench_mm_fusion.py from the pytorch codebase.

Class python

NVSHMEMTileReduceBenchmark Class — pytorch Architecture

Architecture documentation for the NVSHMEMTileReduceBenchmark class in bench_nvshmem_tile_reduce.py from the pytorch codebase.

Class python

Benchmark Class — pytorch Architecture

Architecture documentation for the Benchmark class in benchmark.py from the pytorch codebase.

Class python

TorchvisionBenchmark Class — pytorch Architecture

Architecture documentation for the TorchvisionBenchmark class in benchmark.py from the pytorch codebase.

Class python

RunResult Class — pytorch Architecture

Architecture documentation for the RunResult class in cachebench.py from the pytorch codebase.

Class python

AOTInductorModelCache Class — pytorch Architecture

Architecture documentation for the AOTInductorModelCache class in common.py from the pytorch codebase.

Class python

BenchmarkRunner Class — pytorch Architecture

Architecture documentation for the BenchmarkRunner class in common.py from the pytorch codebase.

Class python

CI Class — pytorch Architecture

Architecture documentation for the CI class in common.py from the pytorch codebase.

Class python

DummyGradScaler Class — pytorch Architecture

Architecture documentation for the DummyGradScaler class in common.py from the pytorch codebase.

Class python

JitTracedCache Class — pytorch Architecture

Architecture documentation for the JitTracedCache class in common.py from the pytorch codebase.

Class python

NativeRTCache Class — pytorch Architecture

Architecture documentation for the NativeRTCache class in common.py from the pytorch codebase.

Class python

Stats Class — pytorch Architecture

Architecture documentation for the Stats class in common.py from the pytorch codebase.

Class python

TimeOutException Class — pytorch Architecture

Architecture documentation for the TimeOutException class in common.py from the pytorch codebase.

Class python

CustomLinear Class — pytorch Architecture

Architecture documentation for the CustomLinear class in dist_util.py from the pytorch codebase.

Class python

MyModule Class — pytorch Architecture

Architecture documentation for the MyModule class in dist_util.py from the pytorch codebase.

Class python

ToyModel Class — pytorch Architecture

Architecture documentation for the ToyModel class in dist_util.py from the pytorch codebase.

Class python

Foo Class — pytorch Architecture

Architecture documentation for the Foo class in dynamo_guard_build.py from the pytorch codebase.

Class python

BwdKernel Class — pytorch Architecture

Architecture documentation for the BwdKernel class in generate_kernels.py from the pytorch codebase.

Class python

FwdKernel Class — pytorch Architecture

Architecture documentation for the FwdKernel class in generate_kernels.py from the pytorch codebase.

Class python

Test Class — pytorch Architecture

Architecture documentation for the Test class in generate_test_torchscripts.py from the pytorch codebase.

Class python

Benchmark Class — pytorch Architecture

Architecture documentation for the Benchmark class in huggingface_llm_models.py from the pytorch codebase.

Class python

TextGenerationBenchmark Class — pytorch Architecture

Architecture documentation for the TextGenerationBenchmark class in huggingface_llm_models.py from the pytorch codebase.

Class python

WhisperBenchmark Class — pytorch Architecture

Architecture documentation for the WhisperBenchmark class in huggingface_llm_models.py from the pytorch codebase.

Class python

HuggingfaceRunner Class — pytorch Architecture

Architecture documentation for the HuggingfaceRunner class in huggingface.py from the pytorch codebase.

Class python

CrossEntropyBackward Class — pytorch Architecture

Architecture documentation for the CrossEntropyBackward class in kernels.py from the pytorch codebase.

Class python

CrossEntropyForward Class — pytorch Architecture

Architecture documentation for the CrossEntropyForward class in kernels.py from the pytorch codebase.

Class python

LayerNormBackward Class — pytorch Architecture

Architecture documentation for the LayerNormBackward class in kernels.py from the pytorch codebase.

Class python

LayerNormForward Class — pytorch Architecture

Architecture documentation for the LayerNormForward class in kernels.py from the pytorch codebase.

Class python

RMSNormBackward Class — pytorch Architecture

Architecture documentation for the RMSNormBackward class in kernels.py from the pytorch codebase.

Class python

RMSNormForward Class — pytorch Architecture

Architecture documentation for the RMSNormForward class in kernels.py from the pytorch codebase.

Class python

SoftmaxBackward Class — pytorch Architecture

Architecture documentation for the SoftmaxBackward class in kernels.py from the pytorch codebase.

Class python

SoftmaxForward Class — pytorch Architecture

Architecture documentation for the SoftmaxForward class in kernels.py from the pytorch codebase.

Class python

MicroBenchmarks Class — pytorch Architecture

Architecture documentation for the MicroBenchmarks class in microbench.py from the pytorch codebase.

Class python

MyModel1 Class — pytorch Architecture

Architecture documentation for the MyModel1 class in microbench.py from the pytorch codebase.

Class python

MyModel2 Class — pytorch Architecture

Architecture documentation for the MyModel2 class in microbench.py from the pytorch codebase.

Class python

NewBatchSampler Class — pytorch Architecture

Architecture documentation for the NewBatchSampler class in samplers_benchmark.py from the pytorch codebase.

Class python

BenchmarkKernel Class — pytorch Architecture

Architecture documentation for the BenchmarkKernel class in utils.py from the pytorch codebase.

Class python

Performance Class — pytorch Architecture

Architecture documentation for the Performance class in utils.py from the pytorch codebase.

Class python

generate_test_torchscripts.py — pytorch Source File

Architecture documentation for generate_test_torchscripts.py, a python file in the pytorch codebase.

File python

configure.py — pytorch Source File

Architecture documentation for configure.py, a python file in the pytorch codebase.

File python

configure.py — pytorch Source File

Architecture documentation for configure.py, a python file in the pytorch codebase.

File python

generate-wrapper.py — pytorch Source File

Architecture documentation for generate-wrapper.py, a python file in the pytorch codebase.

File python

generate_kernels.py — pytorch Source File

Architecture documentation for generate_kernels.py, a python file in the pytorch codebase.

File python

generate_aiter_embedded_hsa.py — pytorch Source File

Architecture documentation for generate_aiter_embedded_hsa.py, a python file in the pytorch codebase.

File python

codegen.py — pytorch Source File

Architecture documentation for codegen.py, a python file in the pytorch codebase.

File python

compare-fastrnn-results.py — pytorch Source File

Architecture documentation for compare-fastrnn-results.py, a python file in the pytorch codebase.

File python

dataloader_benchmark.py — pytorch Source File

Architecture documentation for dataloader_benchmark.py, a python file in the pytorch codebase.

File python

samplers_benchmark.py — pytorch Source File

Architecture documentation for samplers_benchmark.py, a python file in the pytorch codebase.

File python