tensorrt 0.1.0 → 1.0.1
This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
- checksums.yaml +4 -4
- data/ext/tensorrt_rb/tensorrt_rb.cpp +6 -1
- data/lib/tensorrt/version.rb +5 -0
- data/tensorrt.gemspec +4 -2
- metadata +5 -4
checksums.yaml
CHANGED
|
@@ -1,7 +1,7 @@
|
|
|
1
1
|
---
|
|
2
2
|
SHA256:
|
|
3
|
-
metadata.gz:
|
|
4
|
-
data.tar.gz:
|
|
3
|
+
metadata.gz: 536d19669d6667cab73436e472212f6368402c8dcff81e157bd709f550d3bf88
|
|
4
|
+
data.tar.gz: cbf04e8cd9bec907504ae941aad08cd3c1bc05fe0f37eb02684ffbcbf0bd88d8
|
|
5
5
|
SHA512:
|
|
6
|
-
metadata.gz:
|
|
7
|
-
data.tar.gz:
|
|
6
|
+
metadata.gz: b66b517b0e392125a649c1e1dee6d4714e18086da423b7e3fe7713a27eff8fe57e4dbd8a4ce6c1db55c3b0001cdeb6c5d076a95826662c6ffda5bb8d8cfda2ef
|
|
7
|
+
data.tar.gz: 17095661e84ffa65e61f07d726bea0edea211fd6de290371634467fc4da845e68dba711f85402dd91df797c2bd4e41381225820dfbe7fbdb76fe28ed8deb8b57
|
|
@@ -29,9 +29,13 @@ public:
|
|
|
29
29
|
}
|
|
30
30
|
};
|
|
31
31
|
|
|
32
|
+
static TRTLogger& get_logger() {
|
|
33
|
+
static TRTLogger instance;
|
|
34
|
+
return instance;
|
|
35
|
+
}
|
|
36
|
+
|
|
32
37
|
class TRTEngine {
|
|
33
38
|
private:
|
|
34
|
-
TRTLogger logger;
|
|
35
39
|
std::unique_ptr<nvinfer1::IRuntime> runtime;
|
|
36
40
|
std::unique_ptr<nvinfer1::ICudaEngine> engine;
|
|
37
41
|
std::unique_ptr<nvinfer1::IExecutionContext> context;
|
|
@@ -40,6 +44,7 @@ private:
|
|
|
40
44
|
|
|
41
45
|
public:
|
|
42
46
|
TRTEngine(const std::string& engine_path, bool verbose = false) {
|
|
47
|
+
auto& logger = get_logger();
|
|
43
48
|
logger.verbose = verbose;
|
|
44
49
|
|
|
45
50
|
std::ifstream file(engine_path, std::ios::binary);
|
data/tensorrt.gemspec
CHANGED
|
@@ -1,9 +1,11 @@
|
|
|
1
1
|
# frozen_string_literal: true
|
|
2
2
|
|
|
3
|
+
require_relative "lib/tensorrt/version"
|
|
4
|
+
|
|
3
5
|
Gem::Specification.new do |spec|
|
|
4
6
|
spec.name = "tensorrt"
|
|
5
|
-
spec.version =
|
|
6
|
-
spec.author = "Pete
|
|
7
|
+
spec.version = TensorRT::VERSION
|
|
8
|
+
spec.author = "Pete Matsyburka"
|
|
7
9
|
spec.email = ["pete@docuseal.com"]
|
|
8
10
|
|
|
9
11
|
spec.summary = "Minimal TensorRT bindings for Ruby using Rice"
|
metadata
CHANGED
|
@@ -1,13 +1,13 @@
|
|
|
1
1
|
--- !ruby/object:Gem::Specification
|
|
2
2
|
name: tensorrt
|
|
3
3
|
version: !ruby/object:Gem::Version
|
|
4
|
-
version: 0.1
|
|
4
|
+
version: 1.0.1
|
|
5
5
|
platform: ruby
|
|
6
6
|
authors:
|
|
7
|
-
- Pete
|
|
7
|
+
- Pete Matsyburka
|
|
8
8
|
bindir: bin
|
|
9
9
|
cert_chain: []
|
|
10
|
-
date:
|
|
10
|
+
date: 1980-01-02 00:00:00.000000000 Z
|
|
11
11
|
dependencies:
|
|
12
12
|
- !ruby/object:Gem::Dependency
|
|
13
13
|
name: rice
|
|
@@ -79,6 +79,7 @@ files:
|
|
|
79
79
|
- ext/tensorrt_rb/tensorrt_rb.cpp
|
|
80
80
|
- lib/tensorrt.rb
|
|
81
81
|
- lib/tensorrt/cuda.rb
|
|
82
|
+
- lib/tensorrt/version.rb
|
|
82
83
|
- tensorrt.gemspec
|
|
83
84
|
homepage: https://github.com/docusealco/tensorrt-rb
|
|
84
85
|
licenses:
|
|
@@ -98,7 +99,7 @@ required_rubygems_version: !ruby/object:Gem::Requirement
|
|
|
98
99
|
- !ruby/object:Gem::Version
|
|
99
100
|
version: '0'
|
|
100
101
|
requirements: []
|
|
101
|
-
rubygems_version:
|
|
102
|
+
rubygems_version: 4.0.3
|
|
102
103
|
specification_version: 4
|
|
103
104
|
summary: Minimal TensorRT bindings for Ruby using Rice
|
|
104
105
|
test_files: []
|