Commit 30953d8b authored by James Reed's avatar James Reed Committed by Myle Ott
Browse files

Fix for building under clang: specify C++ build and use C++ linkage (#42)

parent c83efd21
......@@ -29,7 +29,7 @@ inline at::Tensor t(at::Type& type, void* i) {
return type.unsafeTensorFromTH(i, true);
}
extern "C" void TemporalConvolutionTBC_forward(
void TemporalConvolutionTBC_forward(
const char* dtype,
void* _input,
void* _output,
......@@ -72,7 +72,7 @@ extern "C" void TemporalConvolutionTBC_forward(
}
}
extern "C" void TemporalConvolutionTBC_backward(
void TemporalConvolutionTBC_backward(
const char* dtype,
void* _dOutput,
void* _dInput,
......
......@@ -42,6 +42,7 @@ conv_tbc = create_extension(
define_macros=[('WITH_CUDA', None)],
with_cuda=True,
extra_compile_args=['-std=c++11'],
source_extension='.cpp',
)
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment