Diagnostics.cpp

Go to the documentation of this file.

 
#include "Diagnostics.hpp"
 
#include <string>
 
#include "llvm/IR/DiagnosticPrinter.h"
#include "llvm/Support/raw_ostream.h"
 
VectorizationDiagnosticHandler::VectorizationDiagnosticHandler()
    : vectorization_failed(false), original_handler(nullptr),
      original_context(nullptr) {}
 
void VectorizationDiagnosticHandler::setOriginalHandler(
    llvm::DiagnosticHandler::DiagnosticHandlerTy handler, void* context) {
    original_handler = handler;
    original_context = context;
}
 
void VectorizationDiagnosticHandler::handleDiagnostic(
    const llvm::DiagnosticInfo& di) {
    bool should_suppress = false;
 
    if (di.getSeverity() == llvm::DS_Remark ||
        di.getSeverity() == llvm::DS_Warning) {
        std::string msg;
        llvm::raw_string_ostream stream(msg);
        llvm::DiagnosticPrinterRawOStream printer(stream);
        di.print(printer);
 
        if (msg.find("loop not vectorized") != std::string::npos) {
            vectorization_failed.store(true);
            should_suppress = true;
        }
    }
 
    // Call original handler for all diagnostics except "loop not vectorized"
    if (!should_suppress && (original_handler != nullptr)) {
        original_handler(&di, original_context);
    }
}
 
bool VectorizationDiagnosticHandler::hasVectorizationFailed() const {
    return vectorization_failed.load();
}
 
void VectorizationDiagnosticHandler::reset() {
    vectorization_failed.store(false);
}
 
void VectorizationDiagnosticHandler::diagnosticHandlerCallback(
    const llvm::DiagnosticInfo* di, void* context) {
    static_cast<VectorizationDiagnosticHandler*>(context)->handleDiagnostic(
        *di);
}