gaspersic/triton
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Core: Fixed operator[int]

Browse Source
This commit is contained in:
Philippe Tillet
2015-08-10 21:55:05 -07:00
parent 0a506fa433
commit ab2dd09309
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
lib/array.cpp
View File

@@ -259,13 +259,13 @@ array_expression array::T() const
scalar array::operator [](int_t idx)
{
assert(nshape()<=1);
return scalar(dtype_, data_, idx);
return scalar(dtype_, data_, start_[0] + ld_*start_[1] + idx);
}
const scalar array::operator [](int_t idx) const
{
assert(nshape()<=1);
return scalar(dtype_, data_, idx);
return scalar(dtype_, data_, start_[0] + ld_*start_[1] + idx);
}