20 lines
641 B
Python
20 lines
641 B
Python
import os
|
|
|
|
from .. import core
|
|
|
|
__path__ = os.path.dirname(os.path.abspath(__file__))
|
|
|
|
|
|
@core.extern
|
|
def globaltimer(_builder=None):
|
|
return core.extern_elementwise("cuda", os.path.join(__path__, "cuda.bc"), [],
|
|
{tuple(): ("globaltimer", core.dtype("int64")),
|
|
}, is_pure=False, _builder=_builder)
|
|
|
|
|
|
@core.extern
|
|
def smid(_builder=None):
|
|
return core.extern_elementwise("cuda", os.path.join(__path__, "cuda.bc"), [],
|
|
{tuple(): ("smid", core.dtype("int32")),
|
|
}, is_pure=True, _builder=_builder)
|