tilelang.carver.arch.cpu

Classes

CPU

Functions

is_cpu_arch(arch)

Module Contents

tilelang.carver.arch.cpu.is_cpu_arch(arch)

参数:

arch (tilelang.carver.arch.arch_base.TileDevice)

返回类型:

bool

class tilelang.carver.arch.cpu.CPU(target)

Bases: tilelang.carver.arch.arch_base.TileDevice

参数:

target (tvm.target.Target)

target

device: tvm.runtime.Device

platform: str = 'CPU'