Skip to content

Tags: gomlx/compute

Tags

v0.1.2

Toggle v0.1.2's commit message

Verified

This commit was created on GitHub.com and signed with GitHub’s verified signature.
Skip tests that use dynamic shapes if not supported. (#22)

v0.1.1

Toggle v0.1.1's commit message
Fix Concatenate axis naming logic to check for dynamic dimension befo…

…re constructing symbolic axis names

v0.1.0

Toggle v0.1.0's commit message

Verified

This commit was created on GitHub.com and signed with GitHub’s verified signature.
Added FusedDense layout (#19)

1. ./compute/fused_ops.go updates:
• Changed FusedDense signature to FusedDense(x, weight, bias Value,
options DenseConfig) (Value, error).
• Defined type DenseConfig struct { Activation ActivationType,
WeightLayout DenseLayout } where DenseLayout can be
DenseLayoutInputOutputs (0, default) or DenseLayoutOutputsInput
      (iota), with documentation.
• Renamed AxesLayout to AttentionAxesLayout and updated constants
(AttentionAxesLayoutBHSD, AttentionAxesLayoutBSHD) along with method
receivers.
2. Refactored AxesLayout usages: • Fixed references across ./compute,
./go-xla, ./onnx-gomlx, and ./gomlx.
3. Code Regeneration in ./compute: • Ran go generate ./... in ./compute
after updating the generators (gobackend_opsregistration and
notimplemented_generator) to recognize DenseConfig, DenseLayout, and
AttentionAxesLayout.
4. Updated gobackend Implementation: • Updated
./compute/internal/gobackend/fusedops/dense.go to handle DenseConfig and
DenseLayoutOutputsInput (contracting the last axis of weight when layout
is DenseLayoutOutputsInput).