level=2 的lod-tensor,在DynamicRNN中对它调用step_input()函数,在CPU上没问题,但是在GPU上报segment fault
Created by: Life-0-1
一个level=2 的lod-tensor ,在DynamicRNN的block中对它调用step_input()函数,在CPU上运行没问题,但是在GPU上报segment fault。代码如下:
drnn = fluid.layers.DynamicRNN()
with drnn.block():
x_ = drnn.step_input(x) # 这里 x 是level=2的lod-tensor
- 版本、环境信息: 1)PaddlePaddle版本:1.3.0_gpu 2)CPU:Xeon 3)GPU:K40 or V100, CUDA 9.0和CUDNN 7.0 4)系统环境:linux,Python 2.7
- 训练信息 1)单卡,K40 OR V100 2)显存信息 3)Operator信息