mirror of https://github.com/Nonannet/copapy.git
optimization for float/int operations added
This commit is contained in:
parent
8fe51a2e45
commit
9f77ef5642
|
|
@ -163,6 +163,8 @@ class variable(Generic[TNum], Net):
|
||||||
@overload
|
@overload
|
||||||
def __mul__(self, other: TVarNumb) -> 'variable[float] | variable[int]': ...
|
def __mul__(self, other: TVarNumb) -> 'variable[float] | variable[int]': ...
|
||||||
def __mul__(self, other: TVarNumb) -> Any:
|
def __mul__(self, other: TVarNumb) -> Any:
|
||||||
|
if self.dtype == 'float' and isinstance(other, int):
|
||||||
|
other = float(other) # Prevent runtime conversion of consts; TODO: add this for other operations
|
||||||
return add_op('mul', [self, other], True)
|
return add_op('mul', [self, other], True)
|
||||||
|
|
||||||
@overload
|
@overload
|
||||||
|
|
|
||||||
Loading…
Reference in New Issue