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

Add MLIR_CAPI_EXPORTED to Triton dialect C bindings

Browse Source 
PiperOrigin-RevId: 604912455
This commit is contained in:
Sergei Lebedev 2024-02-07 02:13:55 -08:00 committed by jax authors
parent 9461b12a3f
commit 01b995e01f
1 changed files with 8 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

13
jaxlib/triton/triton_dialect_capi.h
View File

@ -17,6 +17,7 @@ limitations under the License.
#define JAXLIB_TRITON_TRITON_DIALECT_CAPI_H_
#include "mlir/include/mlir-c/IR.h"
#include "mlir/include/mlir-c/Support.h"
#ifdef __cplusplus
extern "C" {
@ -24,12 +25,14 @@ extern "C" {
MLIR_DECLARE_CAPI_DIALECT_REGISTRATION(Triton, triton);
MlirType mlirTritonPointerTypeGet(MlirType pointeeType, int addressSpace);
bool mlirTritonIsAPointerType(MlirType type);
MlirType mlirTritonPointerTypeGetPointeeType(MlirType pointerType);
MLIR_CAPI_EXPORTED MlirType mlirTritonPointerTypeGet(MlirType pointeeType,
int addressSpace);
MLIR_CAPI_EXPORTED bool mlirTritonIsAPointerType(MlirType type);
MLIR_CAPI_EXPORTED MlirType
mlirTritonPointerTypeGetPointeeType(MlirType pointerType);
MlirAttribute mlirTritonInferReduceOpEncoding(MlirAttribute operandEncoding,
int axis);
MLIR_CAPI_EXPORTED MlirAttribute
mlirTritonInferReduceOpEncoding(MlirAttribute operandEncoding, int axis);
#ifdef __cplusplus
} // extern "C"