diff src/LazyTensors/lazy_tensor_operations.jl @ 1015:4dd3c2312d9e refactor/lazy_tensors

Fix some @boundscheck @inboudns combos
author Jonatan Werpers <jonatan@werpers.com>
date Mon, 21 Mar 2022 13:11:37 +0100
parents f7a718bcb4da
children 5c8c148c56a3
line wrap: on
line diff
--- a/src/LazyTensors/lazy_tensor_operations.jl	Mon Mar 21 09:26:13 2022 +0100
+++ b/src/LazyTensors/lazy_tensor_operations.jl	Mon Mar 21 13:11:37 2022 +0100
@@ -21,9 +21,9 @@
 
 function Base.getindex(ta::LazyTensorApplication{T,R}, I::Vararg{Any,R}) where {T,R}
     @boundscheck checkbounds(ta, Int.(I)...)
-    return apply(ta.t, ta.o, I...)
+    return @inbounds apply(ta.t, ta.o, I...)
 end
-Base.getindex(ta::LazyTensorApplication{T,1} where T, I::CartesianIndex{1}) = ta[Tuple(I)...] # Would otherwise be caught in the previous method.
+Base.@propagate_inbounds @Base.getindex(ta::LazyTensorApplication{T,1} where T, I::CartesianIndex{1}) = ta[Tuple(I)...] # Would otherwise be caught in the previous method.
 Base.size(ta::LazyTensorApplication) = range_size(ta.t)