Skip to content
GitLab
Menu
Projects
Groups
Snippets
Loading...
Help
Help
Support
Community forum
Keyboard shortcuts
?
Submit feedback
Contribute to GitLab
Sign in / Register
Toggle navigation
Menu
Open sidebar
OpenDAS
torch-cluster
Commits
e6812b97
Commit
e6812b97
authored
Apr 07, 2018
by
rusty1s
Browse files
call real kernel
parent
11726e3d
Changes
4
Hide whitespace changes
Inline
Side-by-side
Showing
4 changed files
with
12 additions
and
12 deletions
+12
-12
aten/THC/generic/THCDegree.cuh
aten/THC/generic/THCDegree.cuh
+1
-1
aten/THC/generic/THCPropose.cuh
aten/THC/generic/THCPropose.cuh
+5
-5
aten/THC/generic/THCResponse.cuh
aten/THC/generic/THCResponse.cuh
+5
-5
aten/THCC/generic/THCCGrid.c
aten/THCC/generic/THCCGrid.c
+1
-1
No files found.
aten/THC/generic/THCDegree.cuh
View file @
e6812b97
...
@@ -20,7 +20,7 @@ void THCTensor_(cumDegree)(THCState *state, THCTensor *self, THCudaLongTensor *i
...
@@ -20,7 +20,7 @@ void THCTensor_(cumDegree)(THCState *state, THCTensor *self, THCudaLongTensor *i
real
*
selfData
=
THCTensor_
(
data
)(
state
,
self
);
real
*
selfData
=
THCTensor_
(
data
)(
state
,
self
);
int64_t
*
indexData
=
THCudaLongTensor_data
(
state
,
index
);
int64_t
*
indexData
=
THCudaLongTensor_data
(
state
,
index
);
KERNEL_RUN
(
cumDegreeKernel
,
nEdges
,
selfData
,
indexData
);
KERNEL_
REAL_
RUN
(
cumDegreeKernel
,
nEdges
,
selfData
,
indexData
);
}
}
#endif // THC_GENERIC_FILE
#endif // THC_GENERIC_FILE
aten/THC/generic/THCPropose.cuh
View file @
e6812b97
...
@@ -5,11 +5,11 @@
...
@@ -5,11 +5,11 @@
void
THCTensor_
(
propose
)(
THCState
*
state
,
THCudaLongTensor
*
color
,
THCudaLongTensor
*
prop
,
void
THCTensor_
(
propose
)(
THCState
*
state
,
THCudaLongTensor
*
color
,
THCudaLongTensor
*
prop
,
THCudaLongTensor
*
row
,
THCudaLongTensor
*
col
,
THCTensor
*
weight
,
THCudaLongTensor
*
row
,
THCudaLongTensor
*
col
,
THCTensor
*
weight
,
THCudaLongTensor
*
degree
,
THCudaLongTensor
*
cumDegree
)
{
THCudaLongTensor
*
degree
,
THCudaLongTensor
*
cumDegree
)
{
KERNEL_RUN
(
weightedProposeKernel
,
THCudaLongTensor_nElement
(
state
,
color
),
KERNEL_
REAL_
RUN
(
weightedProposeKernel
,
THCudaLongTensor_nElement
(
state
,
color
),
THCudaLongTensor_data
(
state
,
color
),
THCudaLongTensor_data
(
state
,
prop
),
THCudaLongTensor_data
(
state
,
color
),
THCudaLongTensor_data
(
state
,
prop
),
THCudaLongTensor_data
(
state
,
row
),
THCudaLongTensor_data
(
state
,
col
),
THCudaLongTensor_data
(
state
,
row
),
THCudaLongTensor_data
(
state
,
col
),
THCTensor_
(
data
)(
state
,
weight
),
THCudaLongTensor_data
(
state
,
degree
),
THCTensor_
(
data
)(
state
,
weight
),
THCudaLongTensor_data
(
state
,
degree
),
THCudaLongTensor_data
(
state
,
cumDegree
));
THCudaLongTensor_data
(
state
,
cumDegree
));
}
}
#endif // THC_GENERIC_FILE
#endif // THC_GENERIC_FILE
...
...
aten/THC/generic/THCResponse.cuh
View file @
e6812b97
...
@@ -5,11 +5,11 @@
...
@@ -5,11 +5,11 @@
void
THCTensor_
(
response
)(
THCState
*
state
,
THCudaLongTensor
*
color
,
THCudaLongTensor
*
prop
,
void
THCTensor_
(
response
)(
THCState
*
state
,
THCudaLongTensor
*
color
,
THCudaLongTensor
*
prop
,
THCudaLongTensor
*
row
,
THCudaLongTensor
*
col
,
THCTensor
*
weight
,
THCudaLongTensor
*
row
,
THCudaLongTensor
*
col
,
THCTensor
*
weight
,
THCudaLongTensor
*
degree
,
THCudaLongTensor
*
cumDegree
)
{
THCudaLongTensor
*
degree
,
THCudaLongTensor
*
cumDegree
)
{
KERNEL_RUN
(
weightedResponseKernel
,
THCudaLongTensor_nElement
(
state
,
color
),
KERNEL_
REAL_
RUN
(
weightedResponseKernel
,
THCudaLongTensor_nElement
(
state
,
color
),
THCudaLongTensor_data
(
state
,
color
),
THCudaLongTensor_data
(
state
,
prop
),
THCudaLongTensor_data
(
state
,
color
),
THCudaLongTensor_data
(
state
,
prop
),
THCudaLongTensor_data
(
state
,
row
),
THCudaLongTensor_data
(
state
,
col
),
THCudaLongTensor_data
(
state
,
row
),
THCudaLongTensor_data
(
state
,
col
),
THCTensor_
(
data
)(
state
,
weight
),
THCudaLongTensor_data
(
state
,
degree
),
THCTensor_
(
data
)(
state
,
weight
),
THCudaLongTensor_data
(
state
,
degree
),
THCudaLongTensor_data
(
state
,
cumDegree
));
THCudaLongTensor_data
(
state
,
cumDegree
));
}
}
#endif // THC_GENERIC_FILE
#endif // THC_GENERIC_FILE
aten/THCC/generic/THCCGrid.c
View file @
e6812b97
...
@@ -4,7 +4,7 @@
...
@@ -4,7 +4,7 @@
void
THCCTensor_
(
grid
)(
THCudaLongTensor
*
self
,
THCTensor
*
pos
,
THCTensor
*
size
,
void
THCCTensor_
(
grid
)(
THCudaLongTensor
*
self
,
THCTensor
*
pos
,
THCTensor
*
size
,
THCudaLongTensor
*
count
)
{
THCudaLongTensor
*
count
)
{
/*
THCTensor_(grid)(state, self, pos, size, count);
*/
THCTensor_
(
grid
)(
state
,
self
,
pos
,
size
,
count
);
}
}
#endif
#endif
...
...
Write
Preview
Markdown
is supported
0%
Try again
or
attach a new file
.
Attach a file
Cancel
You are about to add
0
people
to the discussion. Proceed with caution.
Finish editing this message first!
Cancel
Please
register
or
sign in
to comment