Skip to content

CUDA inference extension build by explicitly including torch/nn/functional.h#142

Open
mathew-felix wants to merge 1 commit intomicrosoft:mainfrom
mathew-felix:fix/inference-impl-functional-header
Open

CUDA inference extension build by explicitly including torch/nn/functional.h#142
mathew-felix wants to merge 1 commit intomicrosoft:mainfrom
mathew-felix:fix/inference-impl-functional-header

Conversation

@mathew-felix
Copy link

This PR adds an explicit #include <torch/nn/functional.h> in src/layers/extensions/inference/impl.cpp.

impl.cpp uses torch::nn::functional but currently relies on transitive includes. On my environment, the inference extension build required including this header explicitly.

This is a build-only change and does not modify runtime behavior or model outputs.

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

1 participant