Skip to content
Merged
Show file tree
Hide file tree
Changes from 19 commits
Commits
Show all changes
30 commits
Select commit Hold shift + click to select a range
172be8b
initial
isVoid Dec 3, 2025
7fff78b
adding bfloat16_raw standalone functions
isVoid Dec 4, 2025
0ac190a
use lower_cast from target registry
isVoid Dec 4, 2025
82b8dda
Add API test fp8 hw support
isVoid Dec 4, 2025
eefa9be
instead of fp8_supported, use fp8_accelerated
isVoid Dec 4, 2025
4135d65
add the test
isVoid Dec 4, 2025
a74de5f
add fp8 acceleration info
isVoid Dec 4, 2025
cbeba9f
Merge branch 'main' of github.com:NVIDIA/numba-cuda into fea-fp8
isVoid Dec 5, 2025
4ea11cb
regenerate for numba-cuda ToT
isVoid Dec 5, 2025
1fd63ce
Add Conversion Type Tests
isVoid Dec 5, 2025
0cb3301
add nan tests, update leveraging np.testing
isVoid Dec 5, 2025
5605a88
add existing cvt test
isVoid Dec 5, 2025
b07c226
add cudaroundmode APIs
isVoid Dec 8, 2025
d64acca
Merge remote-tracking branch 'origin' into fea-fp8
isVoid Dec 26, 2025
08c2f7a
update config file and regenerate with ToT Numbast
isVoid Dec 26, 2025
aab8fd6
checkpointing with roundmode tests
isVoid Dec 26, 2025
7a115cb
complete the conversion sat/nosat tests
isVoid Dec 26, 2025
6a30f33
docstring
isVoid Dec 26, 2025
9ea3383
keep sign consistent
isVoid Dec 28, 2025
0c6004e
remove duplicate fp8e5m2 to int8 lowering
isVoid Dec 28, 2025
75afdf9
uncomment skipped tests
isVoid Dec 28, 2025
7bf6763
remove duplicate fp8e5m2 to uint64 conversion
isVoid Dec 28, 2025
09b56d1
add an one-off e4m3 roundtrip debug test in CI to give more insight
isVoid Jan 8, 2026
8632281
Merge branch 'main' into fea-fp8
isVoid Jan 20, 2026
a401e53
Merge branch 'main' of github.com:NVIDIA/numba-cuda into fea-fp8
isVoid Feb 4, 2026
82c9dbc
skip if machine does not support FP8 natively
isVoid Feb 4, 2026
613d266
Merge branch 'fea-fp8' of github.com:isVoid/numba-cuda into fea-fp8
isVoid Feb 4, 2026
feb3a13
remove a debug test file
isVoid Feb 4, 2026
fd46b6d
skip tests on simulators
isVoid Feb 4, 2026
7b83967
Merge branch 'main' of github.com:NVIDIA/numba-cuda into fea-fp8
isVoid Feb 4, 2026
File filter

Filter by extension

Filter by extension


Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
38 changes: 38 additions & 0 deletions configs/cuda_fp8.yml
Original file line number Diff line number Diff line change
@@ -0,0 +1,38 @@
# SPDX-FileCopyrightText: Copyright (c) 2025 NVIDIA CORPORATION & AFFILIATES. All rights reserved.
# SPDX-License-Identifier: BSD-2-Clause
Name: Numba FP8
Version: 0.0.1
GPU Arch:
- sm_90
Entry Point: ./numba_cuda/numba/cuda/include/13/cuda_fp8.h
Clang Include Paths:
- ./numba_cuda/numba/cuda/include/13
File List:
- ./numba_cuda/numba/cuda/include/13/cuda_fp8.h
- ./numba_cuda/numba/cuda/include/13/cuda_fp8.hpp
Exclude: {}
Types:
__nv_fp8_e5m2: Number
__nv_fp8_e4m3: Number
__nv_fp8_e8m0: Number
__nv_fp8x2_e5m2: Type
__nv_fp8x2_e4m3: Type
__nv_fp8x2_e8m0: Type
Data Models:
__nv_fp8_e5m2: PrimitiveModel
__nv_fp8_e4m3: PrimitiveModel
__nv_fp8_e8m0: PrimitiveModel
__nv_fp8x2_e5m2: StructModel
__nv_fp8x2_e4m3: StructModel
__nv_fp8x2_e8m0: StructModel
API Prefix Removal:
Function:
- __nv_
Struct:
- __nv_
Enum:
- __nv_
- __NV_
Skip Prefix: "__internal"
Shim Include Override: "\"cuda_fp8.h\""
Use Separate Registry: True
Loading