tilelang.carver.arch.cpu

類別

CPU

函式

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'