mirrors/rocm_jax
1
0
Fork 0
mirror of https://github.com/ROCm/jax.git synced 2025-04-17 12:26:07 +00:00
Code Issues Packages Projects Releases Wiki Activity

[Mosaic][NFC] Remove static inline for functions in anonymous namespace in source file

Browse Source 
PiperOrigin-RevId: 580974002
This commit is contained in:
Tomás Longeri 2023-11-09 11:31:44 -08:00 committed by jax authors
parent 9b1572c028
commit a23aac5566
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
jaxlib/mosaic/dialect/tpu/integrations/c/tpu_dialect.cc
View File

@ -129,7 +129,7 @@ llvm::ArrayRef<int64_t> mlirTpuI64ArrayRefToLlvmArrayRef(
// We do not use the names wrap/unwrap for MlirTpuValueArray because it
// allocates memory (i.e. they have side effects)
static inline xla::Array<mlir::Value> MlirTpuValueArrayToXlaArray(
xla::Array<mlir::Value> MlirTpuValueArrayToXlaArray(
MlirTpuValueArray arr) {
llvm::ArrayRef<int64_t> shape = mlirTpuI64ArrayRefToLlvmArrayRef(arr.shape);
xla::Array<mlir::Value> res(shape);
@ -139,7 +139,7 @@ static inline xla::Array<mlir::Value> MlirTpuValueArrayToXlaArray(
}
return res;
}
static inline MlirTpuValueArray MlirTpuValueArrayFromXlaArray(
MlirTpuValueArray MlirTpuValueArrayFromXlaArray(
const xla::Array<mlir::Value> &vals) {
int64_t nd = vals.num_dimensions();
int64_t *shape =