bjoernager/rust
bjoernager/rust
1
Fork 0
Code Activity

switch #[cfg(not(llvm_enzyme))] to cfg!(llvm_enzyme)

Browse source
This commit is contained in:
Vayun Biyani 2025-02-27 19:32:30 +05:30
parent cb53e97870
commit c85f038307
2 changed files with 4 additions and 26 deletions
Download patch file Download diff file Expand all files Collapse all files

26
compiler/rustc_builtin_macros/src/autodiff.rs
View file

@ -3,7 +3,6 @@
//! configs (autodiff enabled or disabled), so we have to add cfg's to each import. //! configs (autodiff enabled or disabled), so we have to add cfg's to each import.
//! FIXME(ZuseZ4): Remove this once we have a smarter linter. //! FIXME(ZuseZ4): Remove this once we have a smarter linter.
#[cfg(llvm_enzyme)]
mod llvm_enzyme { mod llvm_enzyme {
use std::str::FromStr; use std::str::FromStr;
use std::string::String; use std::string::String;
@ -130,6 +129,10 @@ mod llvm_enzyme {
meta_item: &ast::MetaItem, meta_item: &ast::MetaItem,
mut item: Annotatable, mut item: Annotatable,
) -> Vec<Annotatable> { ) -> Vec<Annotatable> {
if cfg!(not(llvm_enzyme)) {
ecx.sess.dcx().emit_err(errors::AutoDiffSupportNotBuild { span: meta_item.span });
return vec![item];
}
let dcx = ecx.sess.dcx(); let dcx = ecx.sess.dcx();
// first get the annotable item: // first get the annotable item:
let (sig, is_impl): (FnSig, bool) = match &item { let (sig, is_impl): (FnSig, bool) = match &item {
@ -801,25 +804,4 @@ mod llvm_enzyme {
} }
} }
#[cfg(not(llvm_enzyme))]
mod ad_fallback {
use rustc_ast::ast;
use rustc_expand::base::{Annotatable, ExtCtxt};
use rustc_span::Span;
use crate::errors;
pub(crate) fn expand(
ecx: &mut ExtCtxt<'_>,
_expand_span: Span,
meta_item: &ast::MetaItem,
item: Annotatable,
) -> Vec<Annotatable> {
ecx.sess.dcx().emit_err(errors::AutoDiffSupportNotBuild { span: meta_item.span });
return vec![item];
}
}
#[cfg(not(llvm_enzyme))]
pub(crate) use ad_fallback::expand;
#[cfg(llvm_enzyme)]
pub(crate) use llvm_enzyme::expand; pub(crate) use llvm_enzyme::expand;

4
compiler/rustc_builtin_macros/src/errors.rs
View file

@ -144,10 +144,8 @@ pub(crate) struct AllocMustStatics {
pub(crate) span: Span, pub(crate) span: Span,
} }
#[cfg(llvm_enzyme)]
pub(crate) use autodiff::*; pub(crate) use autodiff::*;
#[cfg(llvm_enzyme)]
mod autodiff { mod autodiff {
use super::*; use super::*;
#[derive(Diagnostic)] #[derive(Diagnostic)]
@ -203,9 +201,7 @@ mod autodiff {
} }
} }
#[cfg(not(llvm_enzyme))]
pub(crate) use ad_fallback::*; pub(crate) use ad_fallback::*;
#[cfg(not(llvm_enzyme))]
mod ad_fallback { mod ad_fallback {
use super::*; use super::*;
#[derive(Diagnostic)] #[derive(Diagnostic)]