[Mlir-commits] [mlir] [mlir][py] Add NVGPU's `TensorMapDescriptorType` in py bindings (PR #88855)
Oleksandr Alex Zinenko
llvmlistbot at llvm.org
Tue Apr 16 01:52:10 PDT 2024
================
@@ -11,13 +11,24 @@
#define MLIR_C_DIALECT_NVGPU_H
#include "mlir-c/IR.h"
+#include "mlir-c/Support.h"
#ifdef __cplusplus
extern "C" {
#endif
MLIR_DECLARE_CAPI_DIALECT_REGISTRATION(NVGPU, nvgpu);
+//===---------------------------------------------------------------------===//
+// TensorMapDescriptorType
+//===---------------------------------------------------------------------===//
+
+MLIR_CAPI_EXPORTED bool mlirTypeIsANVGPUTensorMapDescriptorType(MlirType type);
+
+MLIR_CAPI_EXPORTED MlirType mlirNVGPUTensorMapDescriptorTypeGet(
+ MlirContext ctx, MlirType tensorType, int swizzle, int l2promo, int oob,
----------------
ftynse wrote:
Nit: consider more descriptive names than `oob`.
https://github.com/llvm/llvm-project/pull/88855
More information about the Mlir-commits
mailing list