Lei WangandZhiwen Mo 6d10d1602c [CUDA][Feature] Add packed FP32x2 math intrinsics and auto vectorized support (#1839) ...

* [Feature] Add packed FP32x2 math intrinsics and support for CUDA/HIP

This commit introduces new packed FP32x2 math operations: fadd2, fmul2, and fma2, which leverage PTX instructions on supported architectures. The changes include:

- Definitions and implementations of fadd2, fmul2, and fma2 in both CUDA and HIP codegen files.
- New Python API functions for these operations, ensuring they validate input types and handle fallbacks for unsupported architectures.
- Documentation updates to reflect the new intrinsics in the math language module.

These enhancements improve performance for vectorized floating-point operations in TileLang.

* revert hip changes

* annotate cuda only

---------

Co-authored-by: Zhiwen Mo <zm125@ic.ac.uk>

2026-02-12 12:50:30 +08:00

..

test_cuda_f32x2_intrinsics.py

[CUDA][Feature] Add packed FP32x2 math intrinsics and auto vectorized support (#1839)

2026-02-12 12:50:30 +08:00