Make helion.exc.TensorOperationInWrapper not fire on non-torch ops

stack-info: PR: #131, branch: jansel/stack/27

Author: jansel

helion/_compiler/type_propagation.py

@@ -744,7 +744,9 @@ def to_proxy(arg: TypeInfo) -> object:
                 and input_contains_tensor
                 and output_type.contains_tensor()
             ):
-                if not regexp_allowed_host_ops.search(self.name):
+                if getattr(self.value, "__module__", "").startswith(
+                    "torch"
+                ) and not regexp_allowed_host_ops.search(self.name):
                     warning(exc.TensorOperationInWrapper(self.name))
             return output_type
         except exc.ShapeSpecializingCall:

helion/exc.py

@@ -285,7 +285,9 @@ def report(self) -> str:
 class TensorOperationInWrapper(BaseWarning):
     message = (
         "A tensor operation outside of the `hl.tile` or `hl.grid` loop will not be fused "
-        "in the generated kernel."
+        "in the generated kernel.\n"
+        "Use @helion.kernel(ignore_warnings=[helion.exc.TensorOperationInWrapper]) to suppress this warning.\n"
+        "If this is not a tensor operation, please report this as a bug."
     )