提交 e3f86373 authored 作者: Frederic Bastien's avatar Frederic Bastien

added comment.

上级 5de80af1
...@@ -146,6 +146,13 @@ class GpuDot22Scalar(Op): ...@@ -146,6 +146,13 @@ class GpuDot22Scalar(Op):
gpu_dot22scalar = GpuDot22Scalar() gpu_dot22scalar = GpuDot22Scalar()
class GpuGemm(Op): class GpuGemm(Op):
"""
implement the gemm on the gpu.
..note: This probably don't work correctly for no_inplace gemm.
Need to check al least refcount.
"""
destroy_map = {0:[0]} destroy_map = {0:[0]}
def __str__(self): def __str__(self):
return 'GpuGemm' return 'GpuGemm'
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论