Hetu/python/hetu/gpu_links/MatrixDivideConstLink.py

from __future__ import absolute_import

import ctypes
from .._base import _LIB
from .. import ndarray as _nd


def matrix_elementwise_divide_const(val, in_mat, out_mat, stream=None):
    assert isinstance(in_mat, _nd.NDArray)
    assert isinstance(out_mat, _nd.NDArray)
    _LIB.DLGpuMatrixDivConst(
        ctypes.c_float(val), in_mat.handle, out_mat.handle, stream.handle if stream else None)