提交 047017d6 authored 作者: Frederic's avatar Frederic

Fix compilation crash on windows as nvcc/VS don't support C99

上级 6253b797
......@@ -645,7 +645,10 @@ class Subtensor(Op):
len_subtensor_spec = spec_pos()
is_slice_init = ",".join([str(s) for s in is_slice])
if is_slice:
is_slice_init = "int is_slice[] = {" + ",".join([str(s) for s in is_slice]) + "};"
else:
is_slice_init = "int* is_slice = NULL;"
subtensor_init = "\n".join(init_cmds)
x, = inputs[:1]
......@@ -673,7 +676,7 @@ class Subtensor(Op):
// The subtensor is created by iterating over the dimensions
// and updating stride, shape, and data pointers
int is_slice[] = {%(is_slice_init)s};
%(is_slice_init)s
npy_intp subtensor_spec[%(len_subtensor_spec)s];
%(subtensor_init)s;
int spec_pos = 0; //position in subtensor_spec
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论