提交 3226ffd2 authored 作者: Arnaud Bergeron's avatar Arnaud Bergeron

Fix wrong spelling of _f16_ok

上级 33b437b7
...@@ -215,7 +215,7 @@ class Shape(gof.Op): ...@@ -215,7 +215,7 @@ class Shape(gof.Op):
@note: Non-differentiable. @note: Non-differentiable.
""" """
_fp16_ok = True _f16_ok = True
# Mapping from Type to C code (and version) to use. # Mapping from Type to C code (and version) to use.
# In the C code, the name of the input variable is %(iname)s, # In the C code, the name of the input variable is %(iname)s,
...@@ -310,7 +310,7 @@ class Shape_i(gof.Op): ...@@ -310,7 +310,7 @@ class Shape_i(gof.Op):
@note: Non-differentiable. @note: Non-differentiable.
""" """
_fp16_ok = True _f16_ok = True
# Mapping from Type to C code (and version) to use. # Mapping from Type to C code (and version) to use.
# In the C code, the name of the input variable is %(iname)s, # In the C code, the name of the input variable is %(iname)s,
......
...@@ -21,7 +21,7 @@ from .comp import NVCC_compiler ...@@ -21,7 +21,7 @@ from .comp import NVCC_compiler
class GpuSubtensor(HideC, Subtensor): class GpuSubtensor(HideC, Subtensor):
_fp16_ok = True _f16_ok = True
def make_node(self, x, *inputs): def make_node(self, x, *inputs):
rval = tensor.Subtensor.make_node(self, x, *inputs) rval = tensor.Subtensor.make_node(self, x, *inputs)
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论