|
86 | 86 |
|
87 | 87 | },
|
88 | 88 | exclude = {"openblas_config.h", "cblas.h", "lapacke_config.h", "lapacke_mangling.h", "lapack.h", "lapacke.h", "lapacke_utils.h"},
|
89 |
| - preload = { "asmjit", "fbgemm" } |
| 89 | + link = { "torch" } |
90 | 90 | ),
|
91 | 91 | @Platform(
|
92 | 92 | value = {"linux", "macosx", "windows"},
|
|
105 | 105 | ),
|
106 | 106 | @Platform(
|
107 | 107 | value = {"linux"},
|
108 |
| - link = { "c10", "torch_cpu" } |
| 108 | + preload = { "c10", "torch_cpu" }, |
| 109 | + |
109 | 110 | ),
|
110 | 111 | @Platform(
|
111 | 112 | value = {"macosx"},
|
112 |
| - link = { "c10", "torch_cpu", "iomp5" } |
| 113 | + preload = { "c10", "torch_cpu", "iomp5" } |
113 | 114 | ),
|
114 | 115 | @Platform(
|
115 | 116 | value = "windows",
|
116 |
| - link = { "c10", "torch_cpu", "uv" } |
| 117 | + preload = { "c10", "torch_cpu", "uv" } |
| 118 | + ), |
| 119 | + @Platform( |
| 120 | + value = "linux", |
| 121 | + extension = "-gpu", |
| 122 | + preload = { "c10", "torch_cpu", "c10_cuda", "torch_cuda" } |
117 | 123 | ),
|
| 124 | + @Platform( |
| 125 | + value = "windows", |
| 126 | + extension = "-gpu", |
| 127 | + preload = { "c10", "torch_cpu", "uv", "c10_cuda", "torch_cuda" } |
| 128 | + ) |
118 | 129 | },
|
119 | 130 | target = "org.bytedeco.pytorch",
|
120 | 131 | global = "org.bytedeco.pytorch.global.torch"
|
@@ -1959,11 +1970,11 @@ We need either to put an annotation info on each member, or javaName("@NoOffset
|
1959 | 1970 | "c10::DDPLoggingData::strs_map",
|
1960 | 1971 | "c10::DDPLoggingData::ints_map",
|
1961 | 1972 | "torch::dynamo::autograd::TensorArgs::inputs",
|
| 1973 | + "torch::dynamo::autograd::AutogradCompilerCall::tensor_args", |
1962 | 1974 | "torch::dynamo::autograd::AutogradCompilerCall::all_size_inputs",
|
1963 | 1975 | "torch::dynamo::autograd::AutogradCompilerCall::dyn_size_inputs",
|
1964 | 1976 | "torch::dynamo::autograd::AutogradCompilerCall::node_calls",
|
1965 | 1977 | "torch::dynamo::autograd::AutogradCompilerCall::default_dyn_type",
|
1966 |
| - "torch::dynamo::autograd::AutogradCompilerCall::tensor_args", |
1967 | 1978 | "torch::jit::Object::Property::setter_func",
|
1968 | 1979 | "torch::jit::Object::Property::getter_func",
|
1969 | 1980 | "torch::jit::Object::Property::name",
|
|
0 commit comments