Skip to content

Commit 604b6e4

Browse files
committed
Add zero dimention profile to getitem slice
1 parent c0ae7d3 commit 604b6e4

File tree

1 file changed

+3
-2
lines changed

1 file changed

+3
-2
lines changed

graalpython/com.oracle.graal.python/src/com/oracle/graal/python/builtins/objects/memoryview/MemoryViewBuiltins.java

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -200,12 +200,13 @@ Object getitem(VirtualFrame frame, PMemoryView self, Object index,
200200

201201
@Specialization
202202
Object getitemSlice(PMemoryView self, PSlice slice,
203+
@Shared("zeroDim") @Cached ConditionProfile zeroDimProfile,
203204
@Cached SliceLiteralNode.SliceUnpack sliceUnpack,
204205
@Cached SliceLiteralNode.AdjustIndices adjustIndices,
205206
@Cached MemoryViewNodes.InitFlagsNode initFlagsNode,
206207
@Cached MemoryViewNodes.GetBufferReferences getQueue) {
207208
self.checkReleased(this);
208-
if (self.getDimensions() == 0) {
209+
if (zeroDimProfile.profile(self.getDimensions() == 0)) {
209210
throw raise(TypeError, ErrorMessages.INVALID_INDEXING_OF_0_DIM_MEMORY);
210211
}
211212
int[] shape = self.getBufferShape();
@@ -227,7 +228,7 @@ Object getitemSlice(PMemoryView self, PSlice slice,
227228

228229
@Specialization
229230
Object getitemEllipsis(PMemoryView self, @SuppressWarnings("unused") PEllipsis ellipsis,
230-
@Cached ConditionProfile zeroDimProfile) {
231+
@Shared("zeroDim") @Cached ConditionProfile zeroDimProfile) {
231232
self.checkReleased(this);
232233
if (zeroDimProfile.profile(self.getDimensions() == 0)) {
233234
return self;

0 commit comments

Comments
 (0)