search

openxla / xla

A machine learning compiler for GPUs, CPUs, and ML accelerators
Apache License 2.0
2.64k stars 418 forks source link

Handle gather/scatter batching dims in Get(Gather/Scatter)SizeInChunkRatio for cost analysis. #17955

Closed copybara-service[bot] closed 1 week ago

copybara-service[bot] commented 1 week ago

Handle gather/scatter batching dims in Get(Gather/Scatter)SizeInChunkRatio for cost analysis.