XLA.jl
XLA.jl copied to clipboard
JuliaGPU
Julia on TPUs
"Run in Google Colab" notebook no longer works as the julia-tpu-binaries can no longer be downloaded from: https://storage.googleapis.com/julia-tpu-binaries/julia.v1.1.0-kf.tpu3.x86_64-linux-gnu.tar.gz You get the following error when trying to download: ``` UserProjectAccountProblem User...
Given that there seems to be no activity in issues or commits?
E.g. somebody else may have reset the TPU underneath of us, which is fine since our allocations are dead anyway.
Currently we say things like "", which is informative, but doesn't match Google terminology. We could try printing something like "". We don't get the chip kind information from the...
``` Base.@pure calc_rounds(L; exponent=3) = ceil(UInt32, exponent * log(L) / log(typemax(UInt32))) function shuffle(x::XRTArray) rounds = calc_rounds(length(x)) round_idx = XRTArray(rounds) while round_idx > XRTArray(0) keys = rand(XRTArray{UInt32}, length(x)) x = sort(x,...
Currently it throws the opaque error: ``` AssertionError: isexpr(cond_stmt, :invoke) Stacktrace: [1] outline_if!(::Core.Compiler.IRCode, ::Core.Compiler.OptimizationState, ::Core.Compiler.DomTree, ::Int64, ::Int64) at /home/sabae/.julia/dev/XLA/src/outlining.jl:468 [2] outline_control_flow!(::Core.Compiler.IRCode, ::Core.Compiler.OptimizationState) at /home/sabae/.julia/dev/XLA/src/outlining.jl:503 [3] _compile_to_xla!(::Array{XLA.xla.HloComputationProto,1}, ::XLA.xla.HloComputationProto, ::Core.Compiler.IRCode, ::Core.Compiler.OptimizationState) at...
I am unable to run the example code from TensorFlow.jl without getting a ProcessExitedException() when calling `train.minimize()` in the CoLab environment Input Code: ``` using TensorFlow using MLDatasets import Random...
