ComputeKernels Domain — pytorch Architecture
Hardware-specific implementations of mathematical operations, including QNNPACK and CUDA.
Entity Profile
Dependency Diagram
graph TD domain_ComputeKernels["ComputeKernels"] subdomain_ComputeKernels_ActivationOps["ActivationOps"] domain_ComputeKernels --> subdomain_ComputeKernels_ActivationOps subdomain_ComputeKernels_QuantizedCPU["QuantizedCPU"] domain_ComputeKernels --> subdomain_ComputeKernels_QuantizedCPU subdomain_ComputeKernels_GPUAcceleration["GPUAcceleration"] domain_ComputeKernels --> subdomain_ComputeKernels_GPUAcceleration style domain_ComputeKernels fill:#6366f1,stroke:#818cf8,color:#fff
Relationship Graph
Subdomains
Source Files
- aten/src/ATen/native/quantized/cpu/qnnpack/src/q8gemm_sparse/8x4-packA-sse2.c
- aten/src/ATen/native/quantized/cpu/qnnpack/src/q8gemm_sparse/8x4c1x4-dq-packedA-sse2.c
- aten/src/ATen/native/quantized/cpu/qnnpack/src/q8gemm_sparse/8x4c1x4-dq-packedA-sse2.h
- aten/src/ATen/native/quantized/cpu/qnnpack/src/q8gemm_sparse/8x4c1x4-packed-sse2.h
- aten/src/ATen/native/quantized/cpu/qnnpack/test/add-operator-tester.h
- aten/src/ATen/native/quantized/cpu/qnnpack/test/add.cc
- aten/src/ATen/native/quantized/cpu/qnnpack/test/average-pooling-operator-tester.h
- aten/src/ATen/native/quantized/cpu/qnnpack/test/average-pooling.cc
- aten/src/ATen/native/quantized/cpu/qnnpack/test/avgpool-microkernel-tester.h
- aten/src/ATen/native/quantized/cpu/qnnpack/test/channel-shuffle-operator-tester.h
- aten/src/ATen/native/quantized/cpu/qnnpack/test/channel-shuffle.cc
- aten/src/ATen/native/quantized/cpu/qnnpack/test/clamp-microkernel-tester.h
- aten/src/ATen/native/quantized/cpu/qnnpack/test/clamp-operator-tester.h
- aten/src/ATen/native/quantized/cpu/qnnpack/test/clamp.cc
- aten/src/ATen/native/quantized/cpu/qnnpack/test/convolution-operator-tester.h
- aten/src/ATen/native/quantized/cpu/qnnpack/test/convolution.cc
- aten/src/ATen/native/quantized/cpu/qnnpack/test/deconvolution-operator-tester.h
- aten/src/ATen/native/quantized/cpu/qnnpack/test/deconvolution.cc
- aten/src/ATen/native/quantized/cpu/qnnpack/test/dwconv-microkernel-tester.h
- aten/src/ATen/native/quantized/cpu/qnnpack/test/fully-connected-operator-tester.h
- aten/src/ATen/native/quantized/cpu/qnnpack/test/fully-connected-sparse-operator-tester.h
- aten/src/ATen/native/quantized/cpu/qnnpack/test/fully-connected-sparse.cc
- aten/src/ATen/native/quantized/cpu/qnnpack/test/fully-connected.cc
- aten/src/ATen/native/quantized/cpu/qnnpack/test/gavgpool-microkernel-tester.h
- aten/src/ATen/native/quantized/cpu/qnnpack/test/gemm-block-sparse-microkernel-tester.h
- aten/src/ATen/native/quantized/cpu/qnnpack/test/gemm-microkernel-tester.h
- aten/src/ATen/native/quantized/cpu/qnnpack/src/requantization/gemmlowp-scalar.c
- aten/src/ATen/native/quantized/cpu/qnnpack/src/requantization/gemmlowp-scalar.h
- aten/src/ATen/native/quantized/cpu/qnnpack/src/requantization/gemmlowp-sse.h
- aten/src/ATen/native/quantized/cpu/qnnpack/src/requantization/gemmlowp-sse2.c
- aten/src/ATen/native/quantized/cpu/qnnpack/src/requantization/gemmlowp-sse4.c
- aten/src/ATen/native/quantized/cpu/qnnpack/src/requantization/gemmlowp-ssse3.c
- aten/src/ATen/native/quantized/cpu/qnnpack/test/global-average-pooling-operator-tester.h
- aten/src/ATen/native/quantized/cpu/qnnpack/test/global-average-pooling.cc
- aten/src/ATen/native/quantized/cpu/qnnpack/test/hardsigmoid-operator-tester.h
- aten/src/ATen/native/quantized/cpu/qnnpack/test/hardsigmoid.cc
- aten/src/ATen/native/quantized/cpu/qnnpack/test/hardswish-operator-tester.h
- aten/src/ATen/native/quantized/cpu/qnnpack/test/hardswish.cc
- aten/src/ATen/native/quantized/cpu/qnnpack/test/hgemm.cc
- aten/src/ATen/native/quantized/cpu/qnnpack/test/leaky-relu-operator-tester.h
- aten/src/ATen/native/quantized/cpu/qnnpack/test/leaky-relu.cc
- aten/src/ATen/native/quantized/cpu/qnnpack/test/lut-microkernel-tester.h
- aten/src/ATen/native/quantized/cpu/qnnpack/test/lut-norm-microkernel-tester.h
- aten/src/ATen/native/quantized/cpu/qnnpack/test/max-pooling-operator-tester.h
- aten/src/ATen/native/quantized/cpu/qnnpack/test/max-pooling.cc
- aten/src/ATen/native/quantized/cpu/qnnpack/test/maxpool-microkernel-tester.h
- aten/src/ATen/native/quantized/cpu/qnnpack/test/q8avgpool.cc
- aten/src/ATen/native/quantized/cpu/qnnpack/test/q8conv.cc
- aten/src/ATen/native/quantized/cpu/qnnpack/test/q8dwconv.cc
- aten/src/ATen/native/quantized/cpu/qnnpack/test/q8gavgpool.cc
- aten/src/ATen/native/quantized/cpu/qnnpack/test/q8gemm.cc
- aten/src/ATen/native/quantized/cpu/qnnpack/test/q8gemm_sparse.cc
- aten/src/ATen/native/quantized/cpu/qnnpack/test/q8vadd.cc
- aten/src/ATen/native/quantized/cpu/qnnpack/test/requantization-tester.h
- aten/src/ATen/native/quantized/cpu/qnnpack/test/requantization.cc
- aten/src/ATen/native/quantized/cpu/qnnpack/test/rmax-microkernel-tester.h
- aten/src/ATen/native/quantized/cpu/qnnpack/test/sconv.cc
- aten/src/ATen/native/quantized/cpu/qnnpack/test/sgemm.cc
- aten/src/ATen/native/quantized/cpu/qnnpack/test/sigmoid-operator-tester.h
- aten/src/ATen/native/quantized/cpu/qnnpack/test/sigmoid.cc
- aten/src/ATen/native/quantized/cpu/qnnpack/test/softargmax-operator-tester.h
- aten/src/ATen/native/quantized/cpu/qnnpack/test/softargmax.cc
- aten/src/ATen/native/quantized/cpu/qnnpack/test/tanh-operator-tester.h
- aten/src/ATen/native/quantized/cpu/qnnpack/test/tanh.cc
- aten/src/ATen/native/quantized/cpu/qnnpack/test/test_utils.h
- aten/src/ATen/native/quantized/cpu/qnnpack/test/u8clamp.cc
- aten/src/ATen/native/quantized/cpu/qnnpack/test/u8lut32norm.cc
- aten/src/ATen/native/quantized/cpu/qnnpack/test/u8maxpool.cc
- aten/src/ATen/native/quantized/cpu/qnnpack/test/u8rmax.cc
- aten/src/ATen/native/quantized/cpu/qnnpack/test/vadd-microkernel-tester.h
- aten/src/ATen/native/quantized/cpu/qnnpack/test/x8lut.cc
- aten/src/ATen/native/quantized/cpu/qnnpack/test/x8zip.cc
- aten/src/ATen/native/quantized/cpu/qnnpack/test/zip-microkernel-tester.h
Source
- aten/src/ATen/native/quantized/cpu/qnnpack/src/q8gemm_sparse/8x4-packA-sse2.c
- aten/src/ATen/native/quantized/cpu/qnnpack/src/q8gemm_sparse/8x4c1x4-dq-packedA-sse2.c
- aten/src/ATen/native/quantized/cpu/qnnpack/src/q8gemm_sparse/8x4c1x4-dq-packedA-sse2.h
- aten/src/ATen/native/quantized/cpu/qnnpack/src/q8gemm_sparse/8x4c1x4-packed-sse2.h
- aten/src/ATen/native/quantized/cpu/qnnpack/test/add-operator-tester.h
- aten/src/ATen/native/quantized/cpu/qnnpack/test/add.cc
- aten/src/ATen/native/quantized/cpu/qnnpack/test/average-pooling-operator-tester.h
- aten/src/ATen/native/quantized/cpu/qnnpack/test/average-pooling.cc
- aten/src/ATen/native/quantized/cpu/qnnpack/test/avgpool-microkernel-tester.h
- aten/src/ATen/native/quantized/cpu/qnnpack/test/channel-shuffle-operator-tester.h
- aten/src/ATen/native/quantized/cpu/qnnpack/test/channel-shuffle.cc
- aten/src/ATen/native/quantized/cpu/qnnpack/test/clamp-microkernel-tester.h
- aten/src/ATen/native/quantized/cpu/qnnpack/test/clamp-operator-tester.h
- aten/src/ATen/native/quantized/cpu/qnnpack/test/clamp.cc
- aten/src/ATen/native/quantized/cpu/qnnpack/test/convolution-operator-tester.h
- aten/src/ATen/native/quantized/cpu/qnnpack/test/convolution.cc
- aten/src/ATen/native/quantized/cpu/qnnpack/test/deconvolution-operator-tester.h
- aten/src/ATen/native/quantized/cpu/qnnpack/test/deconvolution.cc
- aten/src/ATen/native/quantized/cpu/qnnpack/test/dwconv-microkernel-tester.h
- aten/src/ATen/native/quantized/cpu/qnnpack/test/fully-connected-operator-tester.h
- aten/src/ATen/native/quantized/cpu/qnnpack/test/fully-connected-sparse-operator-tester.h
- aten/src/ATen/native/quantized/cpu/qnnpack/test/fully-connected-sparse.cc
- aten/src/ATen/native/quantized/cpu/qnnpack/test/fully-connected.cc
- aten/src/ATen/native/quantized/cpu/qnnpack/test/gavgpool-microkernel-tester.h
- aten/src/ATen/native/quantized/cpu/qnnpack/test/gemm-block-sparse-microkernel-tester.h
- aten/src/ATen/native/quantized/cpu/qnnpack/test/gemm-microkernel-tester.h
- aten/src/ATen/native/quantized/cpu/qnnpack/src/requantization/gemmlowp-scalar.c
- aten/src/ATen/native/quantized/cpu/qnnpack/src/requantization/gemmlowp-scalar.h
- aten/src/ATen/native/quantized/cpu/qnnpack/src/requantization/gemmlowp-sse.h
- aten/src/ATen/native/quantized/cpu/qnnpack/src/requantization/gemmlowp-sse2.c
- aten/src/ATen/native/quantized/cpu/qnnpack/src/requantization/gemmlowp-sse4.c
- aten/src/ATen/native/quantized/cpu/qnnpack/src/requantization/gemmlowp-ssse3.c
- aten/src/ATen/native/quantized/cpu/qnnpack/test/global-average-pooling-operator-tester.h
- aten/src/ATen/native/quantized/cpu/qnnpack/test/global-average-pooling.cc
- aten/src/ATen/native/quantized/cpu/qnnpack/test/hardsigmoid-operator-tester.h
- aten/src/ATen/native/quantized/cpu/qnnpack/test/hardsigmoid.cc
- aten/src/ATen/native/quantized/cpu/qnnpack/test/hardswish-operator-tester.h
- aten/src/ATen/native/quantized/cpu/qnnpack/test/hardswish.cc
- aten/src/ATen/native/quantized/cpu/qnnpack/test/hgemm.cc
- aten/src/ATen/native/quantized/cpu/qnnpack/test/leaky-relu-operator-tester.h
- aten/src/ATen/native/quantized/cpu/qnnpack/test/leaky-relu.cc
- aten/src/ATen/native/quantized/cpu/qnnpack/test/lut-microkernel-tester.h
- aten/src/ATen/native/quantized/cpu/qnnpack/test/lut-norm-microkernel-tester.h
- aten/src/ATen/native/quantized/cpu/qnnpack/test/max-pooling-operator-tester.h
- aten/src/ATen/native/quantized/cpu/qnnpack/test/max-pooling.cc
- aten/src/ATen/native/quantized/cpu/qnnpack/test/maxpool-microkernel-tester.h
- aten/src/ATen/native/quantized/cpu/qnnpack/test/q8avgpool.cc
- aten/src/ATen/native/quantized/cpu/qnnpack/test/q8conv.cc
- aten/src/ATen/native/quantized/cpu/qnnpack/test/q8dwconv.cc
- aten/src/ATen/native/quantized/cpu/qnnpack/test/q8gavgpool.cc
- aten/src/ATen/native/quantized/cpu/qnnpack/test/q8gemm.cc
- aten/src/ATen/native/quantized/cpu/qnnpack/test/q8gemm_sparse.cc
- aten/src/ATen/native/quantized/cpu/qnnpack/test/q8vadd.cc
- aten/src/ATen/native/quantized/cpu/qnnpack/test/requantization-tester.h
- aten/src/ATen/native/quantized/cpu/qnnpack/test/requantization.cc
- aten/src/ATen/native/quantized/cpu/qnnpack/test/rmax-microkernel-tester.h
- aten/src/ATen/native/quantized/cpu/qnnpack/test/sconv.cc
- aten/src/ATen/native/quantized/cpu/qnnpack/test/sgemm.cc
- aten/src/ATen/native/quantized/cpu/qnnpack/test/sigmoid-operator-tester.h
- aten/src/ATen/native/quantized/cpu/qnnpack/test/sigmoid.cc
- aten/src/ATen/native/quantized/cpu/qnnpack/test/softargmax-operator-tester.h
- aten/src/ATen/native/quantized/cpu/qnnpack/test/softargmax.cc
- aten/src/ATen/native/quantized/cpu/qnnpack/test/tanh-operator-tester.h
- aten/src/ATen/native/quantized/cpu/qnnpack/test/tanh.cc
- aten/src/ATen/native/quantized/cpu/qnnpack/test/test_utils.h
- aten/src/ATen/native/quantized/cpu/qnnpack/test/u8clamp.cc
- aten/src/ATen/native/quantized/cpu/qnnpack/test/u8lut32norm.cc
- aten/src/ATen/native/quantized/cpu/qnnpack/test/u8maxpool.cc
- aten/src/ATen/native/quantized/cpu/qnnpack/test/u8rmax.cc
- aten/src/ATen/native/quantized/cpu/qnnpack/test/vadd-microkernel-tester.h
- aten/src/ATen/native/quantized/cpu/qnnpack/test/x8lut.cc
- aten/src/ATen/native/quantized/cpu/qnnpack/test/x8zip.cc
- aten/src/ATen/native/quantized/cpu/qnnpack/test/zip-microkernel-tester.h
Frequently Asked Questions
What is the ComputeKernels domain?
The ComputeKernels domain is an architectural grouping in the pytorch codebase. Hardware-specific implementations of mathematical operations, including QNNPACK and CUDA. It contains 73 source files.
What subdomains are in ComputeKernels?
The ComputeKernels domain contains 3 subdomain(s): ActivationOps, GPUAcceleration, QuantizedCPU.
How many files are in ComputeKernels?
The ComputeKernels domain contains 73 source files.
Analyze Your Own Codebase
Get architecture documentation, dependency graphs, and domain analysis for your codebase in minutes.
Try Supermodel Free