nki.isa.memset#

nki.isa.memset(dst, value, engine=engine.unknown, name=None)[source]#

Initialize dst by filling it with a compile-time constant value, using Vector or GpSimd Engine. The memset instruction supports all valid NKI dtypes (see Supported Data Types).

Parameters:

  • dst – destination tile to initialize.

  • value – the constant value to initialize with

  • engine – specify which engine to use for memset: nki.isa.vector_engine or nki.isa.gpsimd_engine ; nki.isa.unknown_engine by default, lets compiler select the best engine for the given input tile shape