This website works better with JavaScript.
Home
Issues
Pull Requests
Milestones
Repositories
Datasets
Forum
实训
竞赛
大数据
应用
Register
Sign In
OpenI
/
MegEngine
Not watched
Unwatch
Watch all
Watch but not notify
1
Star
0
Fork
0
Code
Releases
31
Wiki
Activity
Issues
0
Pull Requests
0
Datasets
Model
Cloudbrain
Browse Source
fix(imperative): fix the size of blob with offset
GitOrigin-RevId:
bb1736dfe1
release-1.5
Megvii Engine Team
huangxinda
4 years ago
parent
787f187e7d
commit
34262d904c
2 changed files
with
2 additions
and
2 deletions
Split View
Diff Options
Show Stats
Download Patch File
Download Diff File
+1
-1
imperative/python/test/unit/jit/test_tracing.py
+1
-1
imperative/src/impl/physical_tensor.cpp
+ 1
- 1
imperative/python/test/unit/jit/test_tracing.py
View File
@@ -583,7 +583,7 @@ def test_trace_advance_indexing(shape_mode):
inputs = {
"x": np.random.randn(5, 5, 5, 5, 5).astype("float32"),
"i":
0
,
"i":
4
,
"j": 2,
"start": 1,
"end": 3,
+ 1
- 1
imperative/src/impl/physical_tensor.cpp
View File
@@ -232,7 +232,7 @@ namespace {
Blob::Blob(const DeviceTensorStorage& s):
m_comp_node{s.comp_node()}, m_storage{s.raw_storage()},
m_size{s.size()} {
m_size{s.size()
+ s.offset()
} {
m_id = next_blob_id++;
BlobManager::inst()->register_blob(this);
}
Write
Preview
Loading…
Cancel
Save