autodiff_zero_cleanup.h

Go to the documentation of this file.

#pragma once
 
#include <mim/def.h>
#include <mim/pass/pass.h>
 
namespace mim::plug::autodiff {
 
/// Replaces remaining zeros (not resolvable) with ⊥.
class AutoDiffZeroCleanup : public RWPass<AutoDiffZeroCleanup, Lam> {
public:
    AutoDiffZeroCleanup(PassMan& man)
        : RWPass(man, "autodiff_zero_cleanup") {}
 
    Ref rewrite(Ref) override;
};
 
} // namespace mim::plug::autodiff