Throw type errors immediately (#202)

The goal here is better error messages.  Before we would defer type
errors in an effort to collect many errors and show them to the user.

Author: jansel

examples/jagged_dense_add.py

@@ -43,7 +43,7 @@ def jagged_dense_add_2d(
     result : dense + jagged, shape (num_rows, N).
     """
     num_rows = y.size(0)
-    assert (*x_offsets.size(),) == (num_rows + 1,)
+    assert x_offsets.size(0) == num_rows + 1
     out = torch.zeros_like(y)
     for tile0 in hl.tile(num_rows):
         starts = x_offsets[tile0]

helion/_compiler/error_reporting.py

@@ -1,12 +1,10 @@
 from __future__ import annotations
 
-from collections import defaultdict
 import functools
 import re
 import sys
 from typing import TYPE_CHECKING
 
-from .. import exc
 from ..exc import Base
 from ..exc import BaseError
 from ..exc import BaseWarning
@@ -16,8 +14,6 @@
     from collections.abc import Callable
     from collections.abc import Sequence
 
-    from .source_location import SourceLocation
-    from .type_propagation import TypeNotAllowedOnDevice
     from helion.runtime.settings import Settings
 
     ErrorOrWarning = BaseError | BaseWarning
@@ -39,9 +35,6 @@ def __init__(self, settings: Settings) -> None:
         self.errors: list[BaseError] = []
         self.warnings: list[BaseWarning] = []
         self.ignores: tuple[type[BaseWarning], ...] = tuple(settings.ignore_warnings)
-        self.type_errors: dict[SourceLocation, list[exc.TypePropagationError]] = (
-            defaultdict(list)
-        )
         self.printed_warning = 0
 
     def add(self, e: Base | type[Base]) -> None:
@@ -61,18 +54,6 @@ def add(self, e: Base | type[Base]) -> None:
         else:
             raise TypeError(f"expected error or warning, got {type(e)}")
 
-    def add_type_error(self, type_info: TypeNotAllowedOnDevice) -> None:
-        """
-        Add a type error to the list of type errors and to the errors list if it's the first occurrence.
-
-        :param type_info: The TypeNotAllowedOnDevice object containing information about the type error.
-        """
-        locations = type_info.locations
-        similar_errors = self.type_errors[locations[0]]
-        similar_errors.append(e := exc.TypePropagationError(type_info, similar_errors))
-        if len(similar_errors) == 1:
-            self.add(e)
-
     def ignore(self, e: type[BaseWarning]) -> None:
         """
         Add a warning type to the ignore list.