EnzymeAD / EnzymeAD/Enzyme-JAX

Weird dus update

Open
#717 2 comments 0 reactions 0 assignees View on GitHub
Dominant language
MLIR
Stars
131
Forks
53
Avg merge
1d 10h
Merged PRs (30d)
193

Description

```
%c_280 = stablehlo.constant dense<7> : tensor
%c_277 = stablehlo.constant dense<8> : tensor
%c_281 = stablehlo.constant dense<0> : tensor

%35 = "enzymexla.comm_region"() ({
%11832 = stablehlo.slice %arg23 [8:9, 8:16, 8:88] : (tensor<20x24x96xf64>) -> tensor<1x8x80xf64>
%11833 = stablehlo.slice %arg23 [0:20, 0:24, 8:88] : (tensor<20x24x96xf64>) -> tensor<20x24x80xf64>
%11834 = stablehlo.dynamic_update_slice %11833, %11832, %c_280, %c_277, %c_281 : (tensor<20x24x80xf64>, tensor<1x8x80xf64>, tensor, tensor, tensor) -> tensor<20x24x80xf64>
stablehlo.return %11834 : tensor<20x24x80xf64>
}) : () -> tensor<20x24x80xf64>

%36 = stablehlo.slice %35 [7:11, 8:16, 0:80] : (tensor<20x24x80xf64>) -> tensor<4x8x80xf64>
%37 = "enzymexla.comm_region"() ({
%11832 = stablehlo.slice %arg24 [8:9, 8:16, 8:88] : (tensor<20x24x96xf64>) -> tensor<1x8x80xf64>
%11833 = stablehlo.slice %arg24 [0:20, 0:24, 8:88] : (tensor<20x24x96xf64>) -> tensor<20x24x80xf64>
%11834 = stablehlo.dynamic_update_slice %11833, %11832, %c_280, %c_277, %c_281 : (tensor<20x24x80xf64>, tensor<1x8x80xf64>, tensor, tensor, tensor) -> tensor<20x24x80xf64>
stablehlo.return %11834 : tensor<20x24x80xf64>
}) : () -> tensor<20x24x80xf64>
```

Contributor guide

No contributing guide indexed for this repository

Assessment

This issue has not been assessed yet.

Get new issues in your inbox

A short digest of beginner-friendly GitHub issues.