diagnostic.h

Go to the documentation of this file.

/*
 * Licensed to the Apache Software Foundation (ASF) under one
 * or more contributor license agreements.  See the NOTICE file
 * distributed with this work for additional information
 * regarding copyright ownership.  The ASF licenses this file
 * to you under the Apache License, Version 2.0 (the
 * "License"); you may not use this file except in compliance
 * with the License.  You may obtain a copy of the License at
 *
 *   http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing,
 * software distributed under the License is distributed on an
 * "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
 * KIND, either express or implied.  See the License for the
 * specific language governing permissions and limitations
 * under the License.
 */
 
#ifndef TVM_IR_DIAGNOSTIC_H_
#define TVM_IR_DIAGNOSTIC_H_
 
#include <tvm/ffi/reflection/registry.h>
#include <tvm/ir/module.h>
 
#include <sstream>
#include <string>
 
namespace tvm {
 
enum class DiagnosticLevel : int {
  kBug = 10,
  kError = 20,
  kWarning = 30,
  kNote = 40,
  kHelp = 50,
};
 
class DiagnosticBuilder;
 
class Diagnostic;
 
class DiagnosticNode : public Object {
 public:
  DiagnosticLevel level;
  Span span;
  ObjectRef loc;
  ffi::String message;
 
  static void RegisterReflection() {
    namespace refl = tvm::ffi::reflection;
    refl::ObjectDef<DiagnosticNode>()
        .def_ro("level", &DiagnosticNode::level)
        .def_ro("span", &DiagnosticNode::span)
        .def_ro("message", &DiagnosticNode::message);
  }
 
  static constexpr TVMFFISEqHashKind _type_s_eq_hash_kind = kTVMFFISEqHashKindTreeNode;
  TVM_FFI_DECLARE_OBJECT_INFO_FINAL("Diagnostic", DiagnosticNode, Object);
};
 
class Diagnostic : public ObjectRef {
 public:
  TVM_DLL Diagnostic(DiagnosticLevel level, Span span, const std::string& message);
 
  static DiagnosticBuilder Bug(Span span);
  static DiagnosticBuilder Error(Span span);
  static DiagnosticBuilder Warning(Span span);
  static DiagnosticBuilder Note(Span span);
  static DiagnosticBuilder Help(Span span);
  // variants uses object location
  static DiagnosticBuilder Bug(ObjectRef loc);
  static DiagnosticBuilder Error(ObjectRef loc);
  static DiagnosticBuilder Warning(ObjectRef loc);
  static DiagnosticBuilder Note(ObjectRef loc);
  static DiagnosticBuilder Help(ObjectRef loc);
  // variants uses object ptr.
  static DiagnosticBuilder Bug(const Object* loc);
  static DiagnosticBuilder Error(const Object* loc);
  static DiagnosticBuilder Warning(const Object* loc);
  static DiagnosticBuilder Note(const Object* loc);
  static DiagnosticBuilder Help(const Object* loc);
 
  TVM_FFI_DEFINE_OBJECT_REF_METHODS_NOTNULLABLE(Diagnostic, ObjectRef, DiagnosticNode);
};
 
class DiagnosticBuilder {
 public:
  DiagnosticLevel level;
 
  SourceName source_name;
 
  Span span;
 
  ObjectRef loc;
 
  template <typename T>
  DiagnosticBuilder& operator<<(const T& val) {  // NOLINT(*)
    stream_ << val;
    return *this;
  }
 
  DiagnosticBuilder() : level(DiagnosticLevel::kError), source_name(), span(Span()) {}
 
  DiagnosticBuilder(const DiagnosticBuilder& builder)
      : level(builder.level), source_name(builder.source_name), span(builder.span) {}
 
  DiagnosticBuilder(DiagnosticLevel level, Span span) : level(level), span(span) {}
 
  DiagnosticBuilder(DiagnosticLevel level, ObjectRef loc) : level(level), loc(loc) {}
 
  operator Diagnostic() { return Diagnostic(this->level, this->span, this->stream_.str()); }
 
 private:
  std::stringstream stream_;
  friend class Diagnostic;
};
 
class DiagnosticContext;
 
class DiagnosticRendererNode : public Object {
 public:
  ffi::TypedFunction<void(DiagnosticContext ctx)> renderer;
 
  static void RegisterReflection() {
    namespace refl = tvm::ffi::reflection;
    refl::ObjectDef<DiagnosticRendererNode>().def_ro("renderer", &DiagnosticRendererNode::renderer);
  }
  TVM_FFI_DECLARE_OBJECT_INFO_FINAL("DiagnosticRenderer", DiagnosticRendererNode, Object);
};
 
class DiagnosticRenderer : public ObjectRef {
 public:
  TVM_DLL DiagnosticRenderer(ffi::TypedFunction<void(DiagnosticContext ctx)> render);
  TVM_DLL DiagnosticRenderer()
      : DiagnosticRenderer(ffi::TypedFunction<void(DiagnosticContext ctx)>()) {}
 
  void Render(const DiagnosticContext& ctx);
 
  DiagnosticRendererNode* operator->() {
    ICHECK(get() != nullptr);
    return static_cast<DiagnosticRendererNode*>(get_mutable());
  }
 
  TVM_FFI_DEFINE_OBJECT_REF_METHODS_NOTNULLABLE(DiagnosticRenderer, ObjectRef,
                                                DiagnosticRendererNode);
};
 
class DiagnosticContextNode : public Object {
 public:
  IRModule module;
 
  ffi::Array<Diagnostic> diagnostics;
 
  DiagnosticRenderer renderer;
 
  static void RegisterReflection() {
    namespace refl = tvm::ffi::reflection;
    refl::ObjectDef<DiagnosticContextNode>()
        .def_ro("module", &DiagnosticContextNode::module)
        .def_ro("diagnostics", &DiagnosticContextNode::diagnostics);
  }
 
  static constexpr TVMFFISEqHashKind _type_s_eq_hash_kind = kTVMFFISEqHashKindTreeNode;
  TVM_FFI_DECLARE_OBJECT_INFO_FINAL("DiagnosticContext", DiagnosticContextNode, Object);
};
 
class DiagnosticContext : public ObjectRef {
 public:
  TVM_DLL DiagnosticContext(const IRModule& module, const DiagnosticRenderer& renderer);
  TVM_DLL static DiagnosticContext Default(const IRModule& source_map);
 
  void Emit(const Diagnostic& diagnostic);
 
  void EmitFatal(const Diagnostic& diagnostic);
 
  void Render();
 
  DiagnosticContextNode* operator->() {
    ICHECK(get() != nullptr);
    return static_cast<DiagnosticContextNode*>(get_mutable());
  }
 
  TVM_FFI_DEFINE_OBJECT_REF_METHODS_NOTNULLABLE(DiagnosticContext, ObjectRef,
                                                DiagnosticContextNode);
};
 
DiagnosticRenderer TerminalRenderer(std::ostream& ostream);
 
}  // namespace tvm
#endif  // TVM_IR_DIAGNOSTIC_H_