Merge pull request #43 from gomlx/dtypes

Changed dtypes imports to github.com/gomlx/gomlx/pkg/core/dtypes.

Author: janpfeifer

go.mod

@@ -7,8 +7,7 @@ require (
 	github.com/daulet/tokenizers v1.23.0
 	github.com/gomlx/exceptions v0.0.3
 	github.com/gomlx/go-huggingface v0.3.1
-	github.com/gomlx/go-xla v0.1.2
-	github.com/gomlx/gomlx v0.26.0-rc1
+	github.com/gomlx/gomlx v0.26.0-rc2
 	github.com/janpfeifer/go-benchmarks v0.1.1
 	github.com/janpfeifer/must v0.2.0
 	github.com/parquet-go/parquet-go v0.24.0
@@ -26,6 +25,7 @@ require (
 	github.com/dustin/go-humanize v1.0.1 // indirect
 	github.com/go-logr/logr v1.4.3 // indirect
 	github.com/gofrs/flock v0.13.0 // indirect
+	github.com/gomlx/go-xla v0.1.2 // indirect
 	github.com/google/uuid v1.6.0 // indirect
 	github.com/klauspost/compress v1.17.9 // indirect
 	github.com/mattn/go-runewidth v0.0.16 // indirect

go.sum

@@ -32,8 +32,8 @@ github.com/gomlx/go-huggingface v0.3.1 h1:kMXA0ecTKywh4xo3WpklCDqUPmyg4ihsyftgCZ
 github.com/gomlx/go-huggingface v0.3.1/go.mod h1:j1cd4gD0A2pwdYHZfkPMYfYr1C+KHCc2YA1kTEBWlTo=
 github.com/gomlx/go-xla v0.1.2 h1:+OQSvPo5VpGdXcyz0bZj5i/sW/k/N1Q9ME60rHcSM7M=
 github.com/gomlx/go-xla v0.1.2/go.mod h1:odQvA8u6ngj7b2rbJjfT/NIN2uQt0/TQvfFpnH25LAs=
-github.com/gomlx/gomlx v0.26.0-rc1 h1:W/AJR3CIhL/VH2tVDZqmtIuXn5tY6vCw7Sp0rgfLAiI=
-github.com/gomlx/gomlx v0.26.0-rc1/go.mod h1:HKSGc4Btb3VntH/BV/UsJYzgTMbLT/Gk69kv95rjruM=
+github.com/gomlx/gomlx v0.26.0-rc2 h1:6mVpej2E7wMJ3zXmm4HNOrmKaKdU9CpSHOwCx6IlgF4=
+github.com/gomlx/gomlx v0.26.0-rc2/go.mod h1:HKSGc4Btb3VntH/BV/UsJYzgTMbLT/Gk69kv95rjruM=
 github.com/google/go-cmp v0.7.0 h1:wk8382ETsv4JYUZwIsn6YpYiWiBsYLSJiTsyBybVuN8=
 github.com/google/go-cmp v0.7.0/go.mod h1:pXiqmnSA92OHEEa9HXL2W4E7lf9JzCmGVUdgjX3N/iU=
 github.com/google/uuid v1.6.0 h1:NIvaJDMOsjHA8n1jAhLSgzrAzy1Hgr+hNrb57e+94F0=

internal/benchmarks/inceptionv3_test.go

@@ -9,7 +9,7 @@ import (
 	"testing"
 
 	"github.com/gomlx/go-huggingface/hub"
-	"github.com/gomlx/go-xla/pkg/types/dtypes"
+	"github.com/gomlx/gomlx/pkg/core/dtypes"
 	. "github.com/gomlx/gomlx/pkg/core/graph"
 	"github.com/gomlx/gomlx/pkg/core/graph/graphtest"
 	"github.com/gomlx/gomlx/pkg/core/shapes"

internal/benchmarks/knights_sbert_test.go

@@ -14,7 +14,7 @@ import (
 	dtok "github.com/daulet/tokenizers"
 	"github.com/gomlx/exceptions"
 	"github.com/gomlx/go-huggingface/hub"
-	"github.com/gomlx/go-xla/pkg/types/dtypes"
+	"github.com/gomlx/gomlx/pkg/core/dtypes"
 	"github.com/gomlx/gomlx/pkg/core/graph"
 	"github.com/gomlx/gomlx/pkg/core/graph/graphtest"
 	"github.com/gomlx/gomlx/pkg/core/shapes"

internal/benchmarks/rob_sentences_test.go

@@ -13,7 +13,7 @@ import (
 	dtok "github.com/daulet/tokenizers"
 	"github.com/gomlx/exceptions"
 	"github.com/gomlx/go-huggingface/hub"
-	"github.com/gomlx/go-xla/pkg/types/dtypes"
+	"github.com/gomlx/gomlx/pkg/core/dtypes"
 	"github.com/gomlx/gomlx/pkg/core/graph"
 	"github.com/gomlx/gomlx/pkg/core/graph/graphtest"
 	"github.com/gomlx/gomlx/pkg/core/shapes"

internal/benchmarks/short_programs_test.go

@@ -19,8 +19,8 @@ import (
 
 	"github.com/chewxy/math32"
 	"github.com/gomlx/exceptions"
-	"github.com/gomlx/go-xla/pkg/types/dtypes"
 	"github.com/gomlx/gomlx/backends"
+	"github.com/gomlx/gomlx/pkg/core/dtypes"
 	"github.com/gomlx/gomlx/pkg/core/graph"
 	"github.com/gomlx/gomlx/pkg/core/graph/graphtest"
 	"github.com/gomlx/gomlx/pkg/core/shapes"

onnx-go.ipynb

@@ -62,7 +62,7 @@
     "\t\"github.com/gomlx/gomlx/ml/context\"\n",
     "\t\"github.com/gomlx/gomlx/ml/data\"\n",
     "\t\"github.com/gomlx/gomlx/types\"\n",
-    "\t\"github.com/gomlx/go-xla/pkg/types/dtypes\"\n",
+    "\t\"github.com/gomlx/gomlx/pkg/core/dtypes\"\n",
     "    \"github.com/gomlx/onnx-gomlx/onnx\"\n",
     "    \"github.com/gomlx/go-huggingface/hub\"\n",
     "    \"github.com/gomlx/go-huggingface/tokenizers\"\n",

onnx/dtypes.go

@@ -2,7 +2,7 @@
 package onnx
 
 import (
-	"github.com/gomlx/go-xla/pkg/types/dtypes"
+	"github.com/gomlx/gomlx/pkg/core/dtypes"
 	"github.com/gomlx/onnx-gomlx/internal/protos"
 	"github.com/pkg/errors"
 )

onnx/dynamicshape.go

@@ -5,7 +5,7 @@ import (
 	"strconv"
 	"strings"
 
-	"github.com/gomlx/go-xla/pkg/types/dtypes"
+	"github.com/gomlx/gomlx/pkg/core/dtypes"
 	"github.com/gomlx/gomlx/pkg/core/shapes"
 	"github.com/gomlx/onnx-gomlx/internal/protos"
 	"github.com/pkg/errors"

onnx/dynamicshape_test.go

@@ -3,7 +3,7 @@ package onnx
 import (
 	"testing"
 
-	"github.com/gomlx/go-xla/pkg/types/dtypes"
+	"github.com/gomlx/gomlx/pkg/core/dtypes"
 	"github.com/gomlx/gomlx/pkg/core/shapes"
 	"github.com/stretchr/testify/require"
 )