Added int8 to Supported traits.

gomlx · Jul 6, 2024 · c9f63f8 · c9f63f8
1 parent bbd150e
commit c9f63f8
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/dtypes/dtypes.go b/dtypes/dtypes.go
@@ -345,7 +345,7 @@ func (dtype DType) IsSupported() bool {
 // Notice Go's `int` type is not portable, since it may translate to dtypes Int32 or Int64 depending
 // on the platform.
 type Supported interface {
-	bool | float32 | float64 | float16.Float16 | int | int32 | int64 | uint8 | uint32 | uint64 | complex64 | complex128
+	bool | float32 | float64 | float16.Float16 | int | int8 | int16 | int32 | int64 | uint8 | uint32 | uint64 | complex64 | complex128
 }
 
 // Number represents the Go numeric types that are supported by graph package.