JuliaGPU/XLA.jl

TensorFlow version mismatch

ViralBShah opened this issue · 0 comments

I started with TF 1.12.

julia> f
f (generic function with 1 method)

julia> @tpu_compile f(XRTArray(rand(Float32,1000,1000)))
1 1 ─ %1 = invoke $(QuoteNode(XLA.HloMap{typeof(+)}()))(+::typeof(+), _2::XRTArray{Float32,(1000, 1000),2}, _2::Vararg{XRTArray{Float32,(1000, 1000),2},N} where N)::XRTArray{Float32,(1000, 1000),2}
  └──      return %1                                                                           │    

57 1 ─ %1 = invoke GenericHloOp{:add}(Float32, ())(_2::XRTArray{Float32,(),0}, _3::Vararg{XRTArray{Float32,(),0},N} where N)::XRTArray{Float32,(),0}
   └──      return %1                                                                                │

ERROR: Tensorflow error: Status: FetchOutputs XRTCompile_3:1: output index too large, must be < 1