Skip to content

Commit 7442e5a

Browse files
authored
misc: remove unused load_cuda_ops function (#1649)
<!-- .github/pull_request_template.md --> ## πŸ“Œ Description `load_cuda_ops` was no longer used and should be removed. ## πŸ” Related Issues <!-- Link any related issues here --> ## πŸš€ Pull Request Checklist Thank you for contributing to FlashInfer! Before we review your pull request, please make sure the following items are complete. ### βœ… Pre-commit Checks - [x] I have installed `pre-commit` by running `pip install pre-commit` (or used your preferred method). - [x] I have installed the hooks with `pre-commit install`. - [x] I have run the hooks manually with `pre-commit run --all-files` and fixed any reported issues. > If you are unsure about how to set up `pre-commit`, see [the pre-commit documentation](https://pre-commit.com/). ## πŸ§ͺ Tests - [x] Tests have been added or updated as needed. - [x] All tests are passing (`unittest`, etc.). ## Reviewer Notes cc @abcdabcd987
1 parent 0828553 commit 7442e5a

File tree

1 file changed

+0
-26
lines changed

1 file changed

+0
-26
lines changed

β€Žflashinfer/jit/core.pyβ€Ž

Lines changed: 0 additions & 26 deletions
Original file line numberDiff line numberDiff line change
@@ -1,7 +1,6 @@
11
import dataclasses
22
import logging
33
import os
4-
import warnings
54
from contextlib import nullcontext
65
from pathlib import Path
76
from typing import List, Optional, Sequence, Union
@@ -248,28 +247,3 @@ def build_jit_specs(
248247
ninja_path = tmpdir / "flashinfer_jit.ninja"
249248
write_if_different(ninja_path, "\n".join(lines))
250249
run_ninja(jit_env.FLASHINFER_JIT_DIR, ninja_path, verbose)
251-
252-
253-
def load_cuda_ops(
254-
name: str,
255-
sources: List[Union[str, Path]],
256-
extra_cflags: Optional[List[str]] = None,
257-
extra_cuda_cflags: Optional[List[str]] = None,
258-
extra_ldflags=None,
259-
extra_include_paths=None,
260-
):
261-
# TODO(lequn): Remove this function and use JitSpec directly.
262-
warnings.warn(
263-
"load_cuda_ops is deprecated. Use JitSpec directly.",
264-
DeprecationWarning,
265-
stacklevel=2,
266-
)
267-
spec = gen_jit_spec(
268-
name=name,
269-
sources=sources,
270-
extra_cflags=extra_cflags,
271-
extra_cuda_cflags=extra_cuda_cflags,
272-
extra_ldflags=extra_ldflags,
273-
extra_include_paths=extra_include_paths,
274-
)
275-
return spec.build_and_load()

0 commit comments

Comments
Β (0)