From 8cf717c9ce46b479580233d9574f937e659561b5 Mon Sep 17 00:00:00 2001 From: Andrew Kane Date: Tue, 16 Jun 2020 15:42:27 -0700 Subject: [PATCH 1/3] Try to fix segfault in tests --- lib/onnxruntime/inference_session.rb | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/lib/onnxruntime/inference_session.rb b/lib/onnxruntime/inference_session.rb index a6d1947..31d90ba 100644 --- a/lib/onnxruntime/inference_session.rb +++ b/lib/onnxruntime/inference_session.rb @@ -385,7 +385,7 @@ def env @@env ||= begin env = ::FFI::MemoryPointer.new(:pointer) check_status api[:CreateEnv].call(3, "Default", env) - at_exit { api[:ReleaseEnv].call(env.read_pointer) } + # at_exit { api[:ReleaseEnv].call(env.read_pointer) } # disable telemetry # https://github.com/microsoft/onnxruntime/blob/master/docs/Privacy.md check_status api[:DisableTelemetryEvents].call(env) From d208f039e3038ff02b13d4807589e03eeb1b9957 Mon Sep 17 00:00:00 2001 From: Andrew Kane Date: Tue, 16 Jun 2020 15:46:36 -0700 Subject: [PATCH 2/3] Run GC --- test/onnxruntime_test.rb | 5 +++++ 1 file changed, 5 insertions(+) diff --git a/test/onnxruntime_test.rb b/test/onnxruntime_test.rb index b6762ea..722ea30 100644 --- a/test/onnxruntime_test.rb +++ b/test/onnxruntime_test.rb @@ -1,6 +1,11 @@ require_relative "test_helper" class OnnxRuntimeTest < Minitest::Test + def setup + puts "GC" + GC.start + end + def test_lib_version assert OnnxRuntime.lib_version end From 51d96ff70fa5977affc3276160637d9739f5f708 Mon Sep 17 00:00:00 2001 From: Andrew Kane Date: Tue, 16 Jun 2020 15:53:18 -0700 Subject: [PATCH 3/3] Set graph optimization level --- lib/onnxruntime/inference_session.rb | 2 +- test/onnxruntime_test.rb | 8 ++++---- 2 files changed, 5 insertions(+), 5 deletions(-) diff --git a/lib/onnxruntime/inference_session.rb b/lib/onnxruntime/inference_session.rb index 31d90ba..1d82f44 100644 --- a/lib/onnxruntime/inference_session.rb +++ b/lib/onnxruntime/inference_session.rb @@ -2,7 +2,7 @@ module OnnxRuntime class InferenceSession attr_reader :inputs, :outputs - def initialize(path_or_bytes, enable_cpu_mem_arena: true, enable_mem_pattern: true, enable_profiling: false, execution_mode: nil, graph_optimization_level: nil, inter_op_num_threads: nil, intra_op_num_threads: nil, log_severity_level: nil, log_verbosity_level: nil, logid: nil, optimized_model_filepath: nil) + def initialize(path_or_bytes, enable_cpu_mem_arena: true, enable_mem_pattern: true, enable_profiling: false, execution_mode: nil, graph_optimization_level: 0, inter_op_num_threads: nil, intra_op_num_threads: nil, log_severity_level: nil, log_verbosity_level: nil, logid: nil, optimized_model_filepath: nil) # session options session_options = ::FFI::MemoryPointer.new(:pointer) check_status api[:CreateSessionOptions].call(session_options) diff --git a/test/onnxruntime_test.rb b/test/onnxruntime_test.rb index 722ea30..ddeb1db 100644 --- a/test/onnxruntime_test.rb +++ b/test/onnxruntime_test.rb @@ -1,10 +1,10 @@ require_relative "test_helper" class OnnxRuntimeTest < Minitest::Test - def setup - puts "GC" - GC.start - end + # def setup + # puts "GC" + # GC.start + # end def test_lib_version assert OnnxRuntime.lib_version