dot.md

Dot Kernel

The dot operator computes the dot product of two vectors.

Mathematical Definition

Given two input vectors x and y, the kernel evaluates

$$ z = x^\top y $$

The dot product is computed by multiplying corresponding elements of x and y, and summing the results to produce a single scalar value z.

Kernel Implementations

• Python Implementation
• PyTorch Implementation
• Triton Implementation
• CuTe Implementation

All backends share the interface:

def dot(x: torch.Tensor, y: torch.Tensor) -> torch.Tensor:
    ...

Testing

See the test suite for the validation harness that exercises every backend.

pytest tests/test_dot.py -s