python - new GPU op in Tensorflow - Segmentation fault error -


i've created new op in tensorflow (tf) , registered gpu. here part of c++ file called pixel_selector.cc:

#include "tensorflow/core/framework/op.h" #include "tensorflow/core/framework/op_kernel.h"  using namespace tensorflow;  register_op("pixelselector")    .input("in: float32")    .input("coord: float32")    .input("stride: int16")    .output("out: float32")  class pixelselectorop : public opkernel {    public:        explicit pixelselectorop(opkernelconstruction* context) : opkernel(context) {}         void compute(opkernelcontext* context) override         {            // grab input tensor            const tensor& input_tensor = context->input(0);            const tensor& input_tensor1 = context->input(1);            const tensor& input_tensor2 = context->input(2);            ...            std::cout << "batch size: " << batch << std::endl;            std::cout << "depth size: " << depth << std::endl;            std::cout << "width size: " << width << std::endl;            std::cout << "height size: " << height << std::endl;            ...            std::cout << "num pixels " << pixels << std::endl;            std::cout << "num coord " << num_coord << std::endl;            ...        } };  register_kernel_builder(name("pixelselector").device(device_gpu), pixelselectorop);

the file has been compiled

tf_inc=$(python -c 'import tensorflow tf; print(tf.sysconfig.get_include())') nvcc -std=c++11 -c -o pixel_selector.o pixel_selector.cc -i $tf_inc -d google_cuda=1 -x cu -xcompiler -fpic g++ -std=c++11 -shared -o pixel_selector.so pixel_selector.o -i $tf_inc -fpic -lcudart -l /usr/local/cuda-8.0/lib64/

but when running in tf session following output:

batch size: batch size: 1111 depth size: 141 width size: 300 height size: 300 num pixels 1 num coord 3 segmentation fault (core dump created)

which looks concurrent access. ideas?


Comments

Post a Comment

Popular posts from this blog

javascript - Clear button on addentry page doesn't work -

this addentry blog page have made scratch , there underlying problems code particularly clear button supposed clear text entered when button clicked on doesn't work have saved html file problem. also 1 other problem submit button works doesn't pass of text index file have written php , reason whenever click on submit button redirects me index page without posting entry have made. <!doctype html> <html> <head> <script> function message() { alert("clear form?"); } </script> <style> body {background-color: beige;} h1 {color: black;} p {color: black;} textarea { width: 50%; height: 150px; padding: 12px 20px; box-sizing: border-box; border: 2px solid #ccc; border-radius: 4px; background-color: white; font-size: 16px; resize: none; } div { padding-left: 20px; } </style> </head> <meta charset="utf-8"> <link rel="stylesheet" href=".
Read more

c# - Selenium Authentication Popup preventing driver close or quit -

i have selenium automated task setup login website retrieve information. it's basic in automation. logging in , clicking 2 links. running issue proving problem. the site in question uses authentication required popup not javascript popup rather regular windows auth popup. have read isn't handled out of box selenium. i using " http://username:password@url.com " site every , site has issues on end prompt me again username/password randomly. have enter username/password twice in row when navigating there hand. since task loops continuously few hours picking needed info figured use webdriverwait verify element present on page. if authentication popup appears, element isn't present. no need care run, start over. the problem having when authentication popup displayed , hit timeout limit want close driver/browser window can start task on again. however, driver.close() , driver.quit() not doing anything. browser remains open , unloaded page still there login p
Read more

tensorflow when input_data MNIST_data , zlib.error: Error -3 while decompressing: invalid block type -

i'm geting started tensorflow,followed tutorials,but cannot download mnist data. import tensorflow tf tensorflow.examples.tutorials.mnist import input_data mnist=input_data.read_data_sets("mnist_data/",one_hot=true) print("download done!") error information: extracting mnist_data/train-images-idx3-ubyte.gz traceback (most recent call last): file "/home/sj/program/tesorflow_try.py", line 6, in <module> mnist=input_data.read_data_sets("mnist_data/",one_hot=true) file "/usr/local/lib/python2.7/dist-packages/tensorflow/contrib/learn/python/learn/datasets/mnist.py", line 213, in read_data_sets train_images = extract_images(f) file "/usr/local/lib/python2.7/dist-packages/tensorflow/contrib/learn/python/learn/datasets/mnist.py", line 60, in extract_images buf = bytestream.read(rows * cols * num_images) file "/usr/lib/python2.7/gzip.py", line 268, in read self._read(readsize) file
Read more