From 4a4e22e0a1f541eb0c25749efe892b150776a8cd Mon Sep 17 00:00:00 2001 From: Melih Elibol <1884921+elibol@users.noreply.github.com> Date: Fri, 30 Oct 2026 15:03:27 -0700 Subject: [PATCH] test(cutile): gate FP8 execution on target capabilities Signed-off-by: Melih Elibol <1884921+elibol@users.noreply.github.com> --- cutile-rs/cutile/tests/dtype_float_ops.rs | 7 +++++++ 1 file changed, 7 insertions(+) diff --git a/cutile-rs/cutile/tests/dtype_float_ops.rs b/cutile-rs/cutile/tests/dtype_float_ops.rs index e5c1ca8efd..b354ea2933 100644 --- a/cutile-rs/cutile/tests/dtype_float_ops.rs +++ b/cutile-rs/cutile/tests/dtype_float_ops.rs @@ -12,6 +12,7 @@ use cuda_core::{f8e4m3fn, f8e5m2}; use cutile::tensor::PartitionMut; use cutile::tile_kernel::{DeviceOp, ToHostVecOp}; +use cutile_ir::requirements::Feature; use half::{bf16, f16}; use std::sync::Arc; @@ -193,6 +194,9 @@ fn add_ones_f64_and_convert_to_f32() { #[test] fn create_ones_f8e4m3fn_and_convert_to_f32() { common::with_test_stack(|| { + if !common::supports_tile_ir(&[Feature::Fp8]) { + return; + } let ones = cutile::api::ones::(&[128]).sync().expect("alloc"); let result_f32: Vec = cutile::api::convert::(Arc::new(ones)) @@ -216,6 +220,9 @@ fn create_ones_f8e4m3fn_and_convert_to_f32() { #[test] fn create_ones_f8e5m2_and_convert_to_f32() { common::with_test_stack(|| { + if !common::supports_tile_ir(&[Feature::Fp8]) { + return; + } let ones = cutile::api::ones::(&[128]).sync().expect("alloc"); let result_f32: Vec = cutile::api::convert::(Arc::new(ones))