Skip to content

Instantly share code, notes, and snippets.

View JackWeiw's full-sized avatar

Wei Tao JackWeiw

4 followers · 7 following
  • East China Normal University
  • ShangHai
  • 15:43 (UTC -12:00)
View GitHub Profile
@JackWeiw
JackWeiw / reprocude.py
Created October 25, 2023 16:04
Script to reproduce dtype mismatch in InjectPTXAsyncCopy PASS
from tvm.script import ir as I
from tvm.script import tir as T
import tvm
from tvm import tir
dtype="float16"
target = tvm.target.Target("nvidia/geforce-rtx-3090")
dev=tvm.cuda(7)
@I.ir_module
class Module:
@JackWeiw
JackWeiw / test.py
Created October 19, 2023 07:57
Below is the script that comes across with Symbolic floormod
import tvm
from tvm import tir
import tvm
from tvm.script import ir as I
from tvm.script import tir as T
from tvm.script import relax as R