import os, sys sys.path.insert(0, os.path.join(os.path.normpath(os.path.join(os.path.dirname(os.path.abspath(__file__)), "..", "..", "..", "..")), "deepinfer", "neuron_ops", "vendor")) _D = os.path.dirname(os.path.abspath(__file__)) os.environ.setdefault("TK_TEST_MERGE_MODE", "overwrite") os.environ.setdefault("TK_BENCHMARK_TESTS_DIR", _D) os.environ.setdefault("TK_BENCHMARK_BASELINES_DIR", os.path.join(_D, "bb")) os.environ.setdefault("TK_BENCHMARK_FAIL_MODE", "none") os.environ.setdefault("TK_GPU_MEM_PROFILES_DIR", os.path.join(_D, "gm")) pytest_plugins = [ "tile_kernels.testing.pytest.benchmark", "tile_kernels.testing.pytest.gpu_mem", "tile_kernels.testing.pytest.random", "tile_kernels.testing.pytest.tilelang_warning", "tile_kernels.testing.pytest.xdist_failfast", ]