• Lei Wang's avatar
    [Bugfix] Fix device type validation for input tensors (#586) · 7b474fbe
    Lei Wang authored
    * Enhancement: Update `pythonic_expr` to accept `tvm.tir.PrimExpr` and improve type handling
    
    - Modified the `pythonic_expr` function to check for `tvm.tir.PrimExpr` type, ensuring proper handling of expressions.
    - Refactored device and dtype checks in `CythonKernelWrapper` for better clarity and error messaging, enhancing robustness in tensor validation.
    
    * Enhancement: Refine `pythonic_expr` function to support additional expression types
    
    - Updated the `pythonic_expr` function to accept `tvm.tir.PrimExpr` and handle both integer and float immediate types, improving expression representation and type handling.
    7b474fbe
utils.py 4.84 KB