fix: specialize

Author: BobTheBuidler

mypyc/irbuild/specialize.py

@@ -1028,6 +1028,6 @@ def specialize_int_to_bytes(builder: IRBuilder, expr: CallExpr, callee: RefExpr)
             )
     # Fallback to generic primitive op
     byteorder_arg = builder.accept(byteorder_expr)
-    return builder.primitive_op(
+    return builder.call_c(
         int_to_bytes_op, [self_arg, length_arg, byteorder_arg, signed_arg], expr.line
     )

mypyc/primitives/int_ops.py

@@ -314,19 +314,9 @@ def int_unary_op(name: str, c_function_name: str) -> PrimitiveDescription:
     error_kind=ERR_NEVER,
 )
 
-# int.to_bytes(length, byteorder)
-method_op(
-    name="to_bytes",
-    arg_types=[int_rprimitive, int_rprimitive, str_rprimitive],
-    extra_int_constants=[(0, bool_rprimitive)],
-    return_type=bytes_rprimitive,
-    c_function_name="CPyTagged_ToBytes",
-    error_kind=ERR_MAGIC,
-)
 
 # int.to_bytes(length, byteorder, signed=...)
-int_to_bytes_op = method_op(
-    name="to_bytes",
+int_to_bytes_op = custom_op(
     arg_types=[int_rprimitive, int_rprimitive, str_rprimitive, bool_rprimitive],
     return_type=bytes_rprimitive,
     c_function_name="CPyTagged_ToBytes",

mypyc/test-data/irbuild-int.test

@@ -216,20 +216,26 @@ def f(x: int) -> bytes:
     return x.to_bytes(2, "big")
 def g(x: int) -> bytes:
     return x.to_bytes(4, "little", signed=True)
+def h(x: int, byteorder: str) -> bytes:
+    return x.to_bytes(8, byteorder)
+
 [out]
 def f(x):
     x :: int
-    r0 :: str
-    r1 :: bytes
+    r0 :: bytes
 L0:
-    r0 = 'big'
-    r1 = CPyTagged_ToBytes(x, 4, r0, 0)
-    return r1
+    r0 = CPyTagged_ToBigEndianBytes(x, 4, 0)
+    return r0
 def g(x):
     x :: int
-    r0 :: str
-    r1 :: bytes
+    r0 :: bytes
 L0:
-    r0 = 'little'
-    r1 = CPyTagged_ToBytes(x, 8, r0, 1)
-    return r1
+    r0 = CPyTagged_ToLittleEndianBytes(x, 8, 1)
+    return r0
+def h(x):
+    x :: int
+    byteorder :: str
+    r0 :: bytes
+L0:
+    r0 = CPyTagged_ToBytes(x, 8, byteorder, 0)
+    return r0