Skip to content

Commit c0525b7

Browse files
authored
[BUG] Fix bug in cast in quantization (#481)
1 parent 04332b7 commit c0525b7

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

python/hidet/graph/ops/quant/symmetric.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -11,7 +11,8 @@
1111
# limitations under the License.
1212
from typing import Union, List
1313
from hidet import ir
14-
from hidet.ir.type import DataType, int32
14+
from hidet.ir.type import DataType
15+
from hidet.ir.dtypes import int32
1516
from hidet.ir.expr import cast, if_then_else
1617
from hidet.ir.compute.primitives import TensorNode, compute
1718
from hidet.ir import primitives as prim

0 commit comments

Comments
 (0)