Merge pull request #24 from gomlx/buffer-bitcast

Added `Buffer.Bitcast`

Author: janpfeifer

cmd/pjrt_installer/main.go

@@ -25,7 +25,7 @@ var (
 
 	flagPlugin, flagPath *string
 	flagVersion          = flag.String("version", "latest",
-		"For PJRT for CPUs, this is the https://github.com/gomlx/pjrt-cpu-binaries release version (e.g.: v0.83.1) "+
+		"For PJRT for CPUs, this is the https://github.com/gomlx/pjrt-cpu-binaries release version (e.g.: v0.98.0) "+
 			"from where to download the plugin. "+
 			"For the CUDA PJRT this is based on the Jax version in https://pypi.org/project/jax/ (e.g.: 0.7.2), "+
 			"which is where it downloads the plugin and Nvidia libraries from. "+

docs/CHANGELOG.md

@@ -25,6 +25,7 @@
 - Installation: 
   - prevent race condition of concurrent (auto-)installations (using flock, so it works cross-processes).
 - Bumped XLA's `pjrt_c_api.h` to version 0.98
+- Added `Buffer.Bitcast`.
 
 # v0.1.4
 

internal/protos/autotune_results/autotune_results.pb.go

@@ -64,6 +64,7 @@ message AutotuneResult {
 
   message GemmKey {
     int64 algorithm = 1;
+    int64 autotune_workspace_size = 2;
   }
 
   // Legacy and unused in new data; superseded by AlgorithmProto.