From 1623541ac54390d132aff51065f0ec3ea7921392 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Jonatan=20K=C5=82osko?= <jonatanklosko@gmail.com>
Date: Thu, 7 Nov 2024 05:30:35 +0100
Subject: [PATCH] Support XLA_ARCHIVE_PATH (#99)

---
 README.md  | 4 ++++
 lib/xla.ex | 7 +++++++
 2 files changed, 11 insertions(+)

diff --git a/README.md b/README.md
index 5d3411b..6a9ab16 100644
--- a/README.md
+++ b/README.md
@@ -65,6 +65,10 @@ Building has a number of dependencies, see *Building from source* below.
 A URL pointing to a specific build of the `.tar.gz` archive. When using this option
 you need to make sure the build matches your OS, CPU architecture and the XLA target.
 
+#### `XLA_ARCHIVE_PATH`
+
+Just like `XLA_ARCHIVE_URL`, but pointing to a local `.tar.gz` archive file.
+
 #### `XLA_CACHE_DIR`
 
 The directory to store the downloaded and built archives in. Defaults to the standard
diff --git a/lib/xla.ex b/lib/xla.ex
index f96222f..94fba37 100644
--- a/lib/xla.ex
+++ b/lib/xla.ex
@@ -38,6 +38,9 @@ defmodule XLA do
         # The archive should have already been built by this point
         archive_path_for_build()
 
+      path = xla_archive_path() ->
+        path
+
       url = xla_archive_url() ->
         path = archive_path_for_external_download(url)
         unless File.exists?(path), do: download_external!(url, path)
@@ -56,6 +59,10 @@ defmodule XLA do
     System.get_env("XLA_BUILD") in ~w(1 true)
   end
 
+  defp xla_archive_path() do
+    System.get_env("XLA_ARCHIVE_PATH")
+  end
+
   defp xla_archive_url() do
     System.get_env("XLA_ARCHIVE_URL")
   end