提交 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): ...@@ -645,7 +645,10 @@ class Subtensor(Op):
len_subtensor_spec = spec_pos() 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) subtensor_init = "\n".join(init_cmds)
x, = inputs[:1] x, = inputs[:1]
...@@ -673,7 +676,7 @@ class Subtensor(Op): ...@@ -673,7 +676,7 @@ class Subtensor(Op):
// The subtensor is created by iterating over the dimensions // The subtensor is created by iterating over the dimensions
// and updating stride, shape, and data pointers // 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]; npy_intp subtensor_spec[%(len_subtensor_spec)s];
%(subtensor_init)s; %(subtensor_init)s;
int spec_pos = 0; //position in subtensor_spec int spec_pos = 0; //position in subtensor_spec
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论