Skip to content

Commit 588d721

Browse files
committed
Simplify aarch64 CpuFeatureLevel with FromStr
1 parent 590d521 commit 588d721

File tree

1 file changed

+3
-5
lines changed

1 file changed

+3
-5
lines changed

src/cpu_features/aarch64.rs

+3-5
Original file line numberDiff line numberDiff line change
@@ -9,9 +9,11 @@
99

1010
use arg_enum_proc_macro::ArgEnum;
1111
use std::env;
12+
use std::str::FromStr;
1213

1314
#[derive(Debug, Copy, Clone, PartialEq, Eq, PartialOrd, ArgEnum)]
1415
pub enum CpuFeatureLevel {
16+
#[arg_enum(alias = "rust")]
1517
NATIVE,
1618
NEON,
1719
}
@@ -31,11 +33,7 @@ impl Default for CpuFeatureLevel {
3133
fn default() -> CpuFeatureLevel {
3234
let detected = CpuFeatureLevel::NEON;
3335
let manual: CpuFeatureLevel = match env::var("RAV1E_CPU_TARGET") {
34-
Ok(feature) => match feature.as_ref() {
35-
"rust" => CpuFeatureLevel::NATIVE,
36-
"neon" => CpuFeatureLevel::NEON,
37-
_ => detected,
38-
},
36+
Ok(feature) => CpuFeatureLevel::from_str(&feature).unwrap_or(detected),
3937
Err(_e) => detected,
4038
};
4139
if manual > detected {

0 commit comments

Comments
 (0)