Skip to content

Commit 5b8d9ff

Browse files
authored
Fix t4a CubeCL xtask package selectors (#9)
1 parent 68365e4 commit 5b8d9ff

3 files changed

Lines changed: 17 additions & 29 deletions

File tree

xtask/src/commands/build.rs

Lines changed: 5 additions & 11 deletions
Original file line numberDiff line numberDiff line change
@@ -14,21 +14,15 @@ pub(crate) fn handle_command(
1414
) -> anyhow::Result<()> {
1515
if args.ci {
1616
// Exclude crates that are not supported on CI
17-
args.exclude
18-
.extend(vec!["cubecl-cuda".to_string(), "cubecl-hip".to_string()]);
17+
args.exclude.extend(vec![
18+
"t4a-cubecl-cuda".to_string(),
19+
"cubecl-hip".to_string(),
20+
]);
1921
}
2022
base_commands::build::handle_command(args.try_into().unwrap(), env, context)?;
2123
// Specific additional commands to test specific features
22-
// burn-wgpu with SPIR-V
2324
helpers::custom_crates_build(
24-
vec!["cubecl-wgpu"],
25-
vec!["--features", "spirv"],
26-
None,
27-
None,
28-
"std with SPIR-V compiler",
29-
)?;
30-
helpers::custom_crates_build(
31-
vec!["cubecl-runtime"],
25+
vec!["t4a-cubecl-runtime"],
3226
vec!["--no-default-features"],
3327
None,
3428
None,

xtask/src/commands/check.rs

Lines changed: 6 additions & 14 deletions
Original file line numberDiff line numberDiff line change
@@ -14,32 +14,24 @@ pub(crate) fn handle_command(
1414
) -> anyhow::Result<()> {
1515
if args.ci {
1616
// Exclude crates that are not supported on CI
17-
args.exclude
18-
.extend(vec!["cubecl-cuda".to_string(), "cubecl-hip".to_string()]);
17+
args.exclude.extend(vec![
18+
"t4a-cubecl-cuda".to_string(),
19+
"cubecl-hip".to_string(),
20+
]);
1921
}
2022
base_commands::check::handle_command(args.try_into().unwrap(), env, context)?;
2123
// Specific additional commands to test specific features
22-
// cubecl-wgpu with SPIR-V
2324
// cubecl-wgpu with exclusive-memory-only
2425
// cubecl-runtime without default features
25-
// Disabled on MacOS see:
26-
#[cfg(not(target_os = "macos"))]
2726
helpers::custom_crates_check(
28-
vec!["cubecl-wgpu"],
29-
vec!["--features", "spirv"],
30-
None,
31-
None,
32-
"std with SPIR-V compiler",
33-
)?;
34-
helpers::custom_crates_check(
35-
vec!["cubecl-wgpu"],
27+
vec!["t4a-cubecl-wgpu"],
3628
vec!["--features", "exclusive-memory-only"],
3729
None,
3830
None,
3931
"std with exclusive_memory_only",
4032
)?;
4133
helpers::custom_crates_check(
42-
vec!["cubecl-runtime"],
34+
vec!["t4a-cubecl-runtime"],
4335
vec!["--no-default-features"],
4436
None,
4537
None,

xtask/src/commands/test.rs

Lines changed: 6 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -15,20 +15,22 @@ pub(crate) fn handle_command(
1515
if args.miri.is_some() {
1616
// miri tests
1717
args.target = Target::Crates;
18-
args.only.extend(vec!["cubecl-common".to_string()]);
18+
args.only.extend(vec!["t4a-cubecl-common".to_string()]);
1919
base_commands::test::handle_command(args.try_into().unwrap(), env, context)?;
2020
} else {
2121
// conventional tests
2222
if args.ci {
2323
// Exclude crates that are not supported on CI
24-
args.exclude
25-
.extend(vec!["cubecl-cuda".to_string(), "cubecl-hip".to_string()]);
24+
args.exclude.extend(vec![
25+
"t4a-cubecl-cuda".to_string(),
26+
"cubecl-hip".to_string(),
27+
]);
2628
}
2729
base_commands::test::handle_command(args.try_into().unwrap(), env, context)?;
2830
// Specific additional commands to test specific features
2931
// cubecl-wgpu with exclusive-memory-only
3032
helpers::custom_crates_tests(
31-
vec!["cubecl-wgpu"],
33+
vec!["t4a-cubecl-wgpu"],
3234
vec!["--features", "exclusive-memory-only", "--lib"],
3335
None,
3436
None,

0 commit comments

Comments
 (0)